diff --git a/third_party/tsl/third_party/gpus/cuda/hermetic/BUILD.tpl b/third_party/tsl/third_party/gpus/cuda/hermetic/BUILD.tpl index da34a336c5a2d..58c4638dd55c3 100644 --- a/third_party/tsl/third_party/gpus/cuda/hermetic/BUILD.tpl +++ b/third_party/tsl/third_party/gpus/cuda/hermetic/BUILD.tpl @@ -89,6 +89,11 @@ cc_library( ], ) +alias( + name = "cuda_runtime", + actual = ":cudart_static", +) + alias( name = "cuda_driver", actual = select({ diff --git a/xla/backends/profiler/gpu/BUILD b/xla/backends/profiler/gpu/BUILD index 3ec55934aaff4..29e31ed5fa083 100644 --- a/xla/backends/profiler/gpu/BUILD +++ b/xla/backends/profiler/gpu/BUILD @@ -134,7 +134,7 @@ cuda_library( visibility = ["//visibility:public"], deps = [ "@local_config_cuda//cuda:cuda_headers", - "@local_config_cuda//cuda:cudart_static", + "@local_config_cuda//cuda:cuda_runtime", "@tsl//tsl/platform:test", ], ) diff --git a/xla/stream_executor/cuda/BUILD b/xla/stream_executor/cuda/BUILD index 109312c6f8cd2..c92a954e824f3 100644 --- a/xla/stream_executor/cuda/BUILD +++ b/xla/stream_executor/cuda/BUILD @@ -1087,7 +1087,7 @@ cc_library( # "//tools/cc_target_os:gce": [], # copybara:uncomment_end_and_comment_begin "//conditions:default": [ - "@local_config_cuda//cuda:cudart_static", # buildcleaner: keep + "@local_config_cuda//cuda:cuda_runtime", # buildcleaner: keep ":cuda_platform", ], }),