10000
We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
xla/stream_executor:cuda_platform
tf_additional_binary_deps
1 parent 42cbd16 commit eeee7f8Copy full SHA for eeee7f8
tensorflow/core/platform/build_config.default.bzl
@@ -1,5 +1,6 @@
1
"""OSS versions of Bazel macros that can't be migrated to TSL."""
2
3
+load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
4
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
5
load(
6
"//tensorflow/core/platform:build_config_root.bzl",
@@ -28,7 +29,9 @@ def tf_additional_binary_deps():
28
29
# core.
30
Label("//tensorflow/core/kernels:lookup_util"),
31
Label("//tensorflow/core/util/tensor_bundle"),
- ] + if_rocm([
32
+ ] + if_cuda([
33
+ Label("@local_xla//xla/stream_executor:cuda_platform"),
34
+ ]) + if_rocm([
35
"@local_xla//xla/stream_executor:rocm_platform",
36
"@local_xla//xla/stream_executor/rocm:rocm_rpath",
37
]) + if_mkl_ml([
0 commit comments