8000 Merge pull request #70357 from tensorflow/r2.16-b24db0b2a85 · IBMZ-Linux-OSS-Python/tensorflow@810f233 · GitHub
[go: up one dir, main page]
Skip to content 8000

Commit 810f233

Browse files
learning-to-playlearning-to-play
authored
Merge pull request tensorflow#70357 from tensorflow/r2.16-b24db0b2a85
r2.16 cherry-pick: b24db0b "Add back `xla/stream_executor:cuda_platform` to `tf_additional_binary_deps`."
2 parents 42cbd16 + eeee7f8 commit 810f233

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

‎tensorflow/core/platform/build_config.default.bzl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
"""OSS versions of Bazel macros that can't be migrated to TSL."""
22

3+
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
34
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
45
load(
56
"//tensorflow/core/platform:build_config_root.bzl",
@@ -28,7 +29,9 @@ def tf_additional_binary_deps():
2829
# core.
2930
Label("//tensorflow/core/kernels:lookup_util"),
3031
Label("//tensorflow/core/util/tensor_bundle"),
31-
] + if_rocm([
32+
] + if_cuda([
33+
Label("@local_xla//xla/stream_executor:cuda_platform"),
34+
]) + if_rocm([
3235
"@local_xla//xla/stream_executor:rocm_platform",
3336
"@local_xla//xla/stream_executor/rocm:rocm_rpath",
3437
]) + if_mkl_ml([

0 commit comments

Comments
 (0)
0