Add back xla/stream_executor:cuda_platform to `tf_additional_binary…

…_deps`. Should fix #63362 Reverts changelist 582804278 PiperOrigin-RevId: 646146985
tensorflow · Jun 24, 2024 · 84938b1 · 84938b1
1 parent 63acb86
commit 84938b1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tensorflow/core/platform/build_config.default.bzl b/tensorflow/core/platform/build_config.default.bzl
@@ -1,5 +1,6 @@
 """OSS versions of Bazel macros that can't be migrated to TSL."""
 
+load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
 load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
 load(
     "@local_xla//xla/tsl:tsl.bzl",
@@ -28,7 +29,9 @@ def tf_additional_binary_deps():
         # core.
         Label("//tensorflow/core/kernels:lookup_util"),
         Label("//tensorflow/core/util/tensor_bundle"),
-    ] + if_rocm([
+    ] + if_cuda([
+        Label("@local_xla//xla/stream_executor:cuda_platform"),
+    ]) + if_rocm([
         "@local_xla//xla/stream_executor:rocm_platform",
         "@local_xla//xla/stream_executor/rocm:rocm_rpath",
     ]) + if_mkl_ml([