experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Using "is_cuda_is_configured_compat" to select the cuda libraries.

Browse Source 
PiperOrigin-RevId: 307728709
Change-Id: I90edb98ca5a2eb641f27e3fa54d6a6cbcaf8420e
This commit is contained in:
A. Unique TensorFlower 2020-04-21 19:27:49 -07:00 committed by TensorFlower Gardener
parent ab0cbb3cc0
commit 4b700752f3
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/profiler/internal/gpu/BUILD
View File

@ -91,8 +91,9 @@ tf_cuda_library(
deps = [ deps = [
"//tensorflow/core:lib", "//tensorflow/core:lib",
"//tensorflow/core:platform_base", "//tensorflow/core:platform_base",
] + if_cuda_is_configured_compat([
"//tensorflow/stream_executor/cuda:cupti_stub", "//tensorflow/stream_executor/cuda:cupti_stub",
], ]),
) )
tf_cuda_library( tf_cuda_library(
@ -103,8 +104,9 @@ tf_cuda_library(
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":cupti_interface", ":cupti_interface",
] + if_cuda_is_configured_compat([
"//tensorflow/stream_executor/cuda:cupti_stub", "//tensorflow/stream_executor/cuda:cupti_stub",
], ]),
) )
tf_cuda_library( tf_cuda_library(