diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-12 15:40:29 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-12 15:42:51 -0700 |
commit | c03bd90c5c89856e53a33f9bae9130237abd3914 (patch) | |
tree | 039d7016e820119347c4427509f467120c58a5b1 | |
parent | 9a1f684b15d3c6011505425bdcc71fe9f986f388 (diff) |
Automated g4 rollback of changelist 196349902
PiperOrigin-RevId: 196387391
-rw-r--r-- | tensorflow/contrib/fused_conv/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/core/grappler/clusters/BUILD | 3 | ||||
-rw-r--r-- | tensorflow/core/grappler/costs/BUILD | 3 | ||||
-rw-r--r-- | tensorflow/core/kernels/BUILD | 4 | ||||
-rw-r--r-- | third_party/gpus/cuda/BUILD.tpl | 9 |
5 files changed, 2 insertions, 19 deletions
diff --git a/tensorflow/contrib/fused_conv/BUILD b/tensorflow/contrib/fused_conv/BUILD index 0f0813c07f..0eb6889db1 100644 --- a/tensorflow/contrib/fused_conv/BUILD +++ b/tensorflow/contrib/fused_conv/BUILD @@ -75,7 +75,6 @@ tf_kernel_library( "//tensorflow/core/kernels:gpu_util_hdrs", "//tensorflow/core/kernels:ops_util_hdrs", "//third_party/eigen3", - "@local_config_cuda//cuda:cudnn_header", ], alwayslink = 1, ) @@ -95,7 +94,6 @@ tf_custom_op_library( "//tensorflow/core/kernels:conv_ops_gpu_hdrs", "//tensorflow/core/kernels:gpu_util_hdrs", "//tensorflow/core/kernels:ops_util_hdrs", - "@local_config_cuda//cuda:cudnn_header", ], ) diff --git a/tensorflow/core/grappler/clusters/BUILD b/tensorflow/core/grappler/clusters/BUILD index d0b2cf01be..30c6126fbb 100644 --- a/tensorflow/core/grappler/clusters/BUILD +++ b/tensorflow/core/grappler/clusters/BUILD @@ -20,9 +20,6 @@ tf_cuda_library( name = "utils", srcs = ["utils.cc"], hdrs = ["utils.h"], - cuda_deps = [ - "@local_config_cuda//cuda:cudnn_header", - ], visibility = ["//visibility:public"], deps = [ "//third_party/eigen3", diff --git a/tensorflow/core/grappler/costs/BUILD b/tensorflow/core/grappler/costs/BUILD index b054068299..35f11eac29 100644 --- a/tensorflow/core/grappler/costs/BUILD +++ b/tensorflow/core/grappler/costs/BUILD @@ -129,9 +129,6 @@ tf_cuda_library( name = "utils", srcs = ["utils.cc"], hdrs = ["utils.h"], - cuda_deps = [ - "@local_config_cuda//cuda:cudnn_header", - ], visibility = ["//visibility:public"], deps = [ "//third_party/eigen3", diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 0263967056..3fb03cd5bd 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -3301,7 +3301,7 @@ tf_kernel_library( "//tensorflow/core:nn_ops_op_lib", ] + if_cuda([ "@cub_archive//:cub", - "@local_config_cuda//cuda:cudnn_header", + "@local_config_cuda//cuda:cudnn", ]), ) @@ -3320,7 +3320,7 @@ tf_kernel_library( "//tensorflow/core:lib", "//tensorflow/core:nn_ops_op_lib", ] + if_cuda([ - "@local_config_cuda//cuda:cudnn_header", + "@local_config_cuda//cuda:cudnn", ]), ) diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl index f6b497f813..2a37c65bc7 100644 --- a/third_party/gpus/cuda/BUILD.tpl +++ b/third_party/gpus/cuda/BUILD.tpl @@ -128,15 +128,6 @@ cc_library( ) cc_library( - name = "cudnn_header", - includes = [ - ".", - "cuda/include", - ], - visibility = ["//visibility:public"], -) - -cc_library( name = "cufft", srcs = ["cuda/lib/%{cufft_lib}"], data = ["cuda/lib/%{cufft_lib}"], |