[go: nahoru, domu]

Skip to content

Commit

Permalink
Make :redzone_allocator_kernel_cuda a cc_library
Browse files Browse the repository at this point in the history
It used to be a `gpu_kernel_library` which means it gets compiled
for device using NVCC in OSS. But the sources files don't contain
any CUDA code, so there is no need for that.

openxla/xla#13460 reports that this behaviour
even fails with some versions of NVCC because NVCC can't deal
with C++ template magic being pulled in by Abseil.

PiperOrigin-RevId: 644685844
  • Loading branch information
beckerhe authored and tensorflower-gardener committed Jun 19, 2024
1 parent ca60e39 commit a2db2af
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion third_party/xla/xla/stream_executor/gpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -491,7 +491,7 @@ gpu_only_cc_library(
]),
)

gpu_kernel_library(
cc_library(
name = "redzone_allocator_kernel_cuda",
srcs = [
"redzone_allocator_kernel.h",
Expand Down

0 comments on commit a2db2af

Please sign in to comment.