diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index c782a9e4528a1..a35ab7cab957a 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -960,6 +960,7 @@ cc_library( "@tsl//tsl/platform:tensor_float_32_hdr_lib", ]) + if_rocm_is_configured([ #keep sorted + "//xla/stream_executor/platform:dso_loader", "//xla/stream_executor/rocm:amdhipblaslt_plugin", "//xla/stream_executor/rocm:hipblas_lt_header", "@local_config_rocm//rocm:rocm_headers", diff --git a/xla/stream_executor/cuda/BUILD b/xla/stream_executor/cuda/BUILD index 473816c37adaf..60d25bec6c369 100644 --- a/xla/stream_executor/cuda/BUILD +++ b/xla/stream_executor/cuda/BUILD @@ -1079,6 +1079,7 @@ cc_library( "//xla/stream_executor:scratch_allocator", "//xla/stream_executor/cuda:cuda_platform_id", "//xla/stream_executor/host:host_platform_id", + "//xla/stream_executor/platform:dso_loader", "//xla/stream_executor/rocm:rocm_platform_id", ] + if_google( select({ diff --git a/xla/stream_executor/sycl/BUILD b/xla/stream_executor/sycl/BUILD index 03afcd832c840..6f3e47845462d 100644 --- a/xla/stream_executor/sycl/BUILD +++ b/xla/stream_executor/sycl/BUILD @@ -86,5 +86,6 @@ cc_library( "//xla/stream_executor:dnn", "//xla/stream_executor:platform_manager", "//xla/stream_executor:scratch_allocator", + "//xla/stream_executor/platform:dso_loader", ] + if_static([":all_runtime"]), )