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

Return select() statement from if_cuda_is_configured().

Browse Source 
PiperOrigin-RevId: 247196015
This commit is contained in:
A. Unique TensorFlower 2019-05-08 05:35:44 -07:00 committed by TensorFlower Gardener
parent d58b53e19c
commit 66b193faee
4 changed files with 49 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/BUILD
View File

@ -3326,7 +3326,7 @@ tf_cuda_library(
name = "device_tracer",
srcs = tf_additional_device_tracer_srcs(),
copts = tf_copts(),
cuda_deps = if_cuda_is_configured(tf_additional_cupti_wrapper_deps() + tf_additional_device_tracer_cuda_deps()),
cuda_deps = tf_additional_cupti_wrapper_deps() + tf_additional_device_tracer_cuda_deps(),
visibility = [
"//tensorflow:internal",
],

60
tensorflow/stream_executor/cuda/BUILD
View File

@ -198,12 +198,22 @@ cc_library(
]),
)
alias(
name = "cublas_lib",
actual = if_static(
"@local_config_cuda//cuda:cublas",
":cublas_stub",
),
visibility = ["//visibility:private"],
)
cc_library(
name = "cublas_plugin",
srcs = if_cuda_is_configured(["cuda_blas.cc"]),
hdrs = if_cuda_is_configured(["cuda_blas.h"]),
visibility = ["//visibility:public"],
deps = if_cuda_is_configured([
":cublas_lib",
":cuda_activation",
":cuda_gpu_executor",
":cuda_platform_id",
@ -223,10 +233,7 @@ cc_library(
"//tensorflow/stream_executor/gpu:gpu_helpers_header",
"//tensorflow/stream_executor/lib",
"//tensorflow/stream_executor/platform",
] + if_static(
["@local_config_cuda//cuda:cublas"],
[":cublas_stub"],
)) + [
]) + [
"@com_google_absl//absl/strings:str_format",
"@com_google_absl//absl/synchronization",
],
@ -244,6 +251,15 @@ cc_library(
]),
)
alias(
name = "cufft_lib",
actual = if_static(
"@local_config_cuda//cuda:cufft",
":cufft_stub",
),
visibility = ["//visibility:private"],
)
cc_library(
name = "cufft_plugin",
srcs = if_cuda_is_configured(["cuda_fft.cc"]),
@ -255,6 +271,7 @@ cc_library(
":cuda_platform_id",
":cuda_stream",
":cuda_helpers",
":cufft_lib",
"@local_config_cuda//cuda:cuda_headers",
"//tensorflow/stream_executor:event",
"//tensorflow/stream_executor:fft",
@ -264,10 +281,7 @@ cc_library(
"//tensorflow/stream_executor/lib",
"//tensorflow/stream_executor/platform",
"//tensorflow/stream_executor/platform:dso_loader",
] + if_static(
["@local_config_cuda//cuda:cufft"],
[":cufft_stub"],
)),
]),
alwayslink = True,
)
@ -282,6 +296,15 @@ cc_library(
]),
)
alias(
name = "cudnn_lib",
actual = if_static(
"@local_config_cuda//cuda:cudnn",
":cudnn_stub",
),
visibility = ["//visibility:private"],
)
cc_library(
name = "cudnn_plugin",
srcs = if_cuda_is_configured(["cuda_dnn.cc"]),
@ -296,6 +319,7 @@ cc_library(
":cuda_stream",
":cuda_timer",
":cudnn_version",
":cudnn_lib",
"@com_google_absl//absl/strings",
"//third_party/eigen3",
"@local_config_cuda//cuda:cuda_headers",
@ -309,10 +333,7 @@ cc_library(
"//tensorflow/stream_executor:temporary_device_memory",
"//tensorflow/stream_executor/lib",
"//tensorflow/stream_executor/platform",
]) + tf_additional_cudnn_plugin_deps() + if_cuda_is_configured(if_static(
["@local_config_cuda//cuda:cudnn"],
[":cudnn_stub"],
)) + ["@com_google_absl//absl/synchronization"],
]) + tf_additional_cudnn_plugin_deps() + ["@com_google_absl//absl/synchronization"],
alwayslink = True,
)
@ -327,6 +348,15 @@ cc_library(
]),
)
alias(
name = "curand_lib",
actual = if_static(
"@local_config_cuda//cuda:curand",
":curand_stub",
),
visibility = ["//visibility:private"],
)
cc_library(
name = "curand_plugin",
srcs = if_cuda_is_configured(["cuda_rng.cc"]),
@ -337,6 +367,7 @@ cc_library(
":cuda_platform_id",
":cuda_stream",
":cuda_helpers",
":curand_lib",
"@local_config_cuda//cuda:cuda_headers",
"//tensorflow/stream_executor:event",
"//tensorflow/stream_executor:plugin_registry",
@ -346,10 +377,7 @@ cc_library(
"//tensorflow/stream_executor/lib",
"//tensorflow/stream_executor/platform",
"//tensorflow/stream_executor/platform:dso_loader",
] + if_static(
["@local_config_cuda//cuda:curand"],
[":curand_stub"],
)),
]),
alwayslink = True,
)

4
third_party/gpus/cuda/build_defs.bzl.tpl vendored
View File

@ -29,5 +29,5 @@ def if_cuda_is_configured(x):
--config=cuda. Used to allow non-CUDA code to depend on CUDA libraries.
"""
if cuda_is_configured():
return x
return []
return select({"//conditions:default": x})
return select({"//conditions:default": []})

4
third_party/toolchains/preconfig/ubuntu14.04/cuda10.0-cudnn7/cuda/build_defs.bzl vendored
View File

@ -27,5 +27,5 @@ def if_cuda_is_configured(x):
--config=cuda. Used to allow non-CUDA code to depend on CUDA libraries.
"""
if cuda_is_configured():
return x
return []
return select({"//conditions:default": x})
return select({"//conditions:default": []})