diff options
| -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, 19 insertions, 2 deletions
diff --git a/tensorflow/contrib/fused_conv/BUILD b/tensorflow/contrib/fused_conv/BUILD index 0eb6889db1..0f0813c07f 100644 --- a/tensorflow/contrib/fused_conv/BUILD +++ b/tensorflow/contrib/fused_conv/BUILD @@ -75,6 +75,7 @@ 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, ) @@ -94,6 +95,7 @@ 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 30c6126fbb..d0b2cf01be 100644 --- a/tensorflow/core/grappler/clusters/BUILD +++ b/tensorflow/core/grappler/clusters/BUILD @@ -20,6 +20,9 @@ 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 35f11eac29..b054068299 100644 --- a/tensorflow/core/grappler/costs/BUILD +++ b/tensorflow/core/grappler/costs/BUILD @@ -129,6 +129,9 @@ 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 3fb03cd5bd..0263967056 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", + "@local_config_cuda//cuda:cudnn_header", ]), ) @@ -3320,7 +3320,7 @@ tf_kernel_library( "//tensorflow/core:lib", "//tensorflow/core:nn_ops_op_lib", ] + if_cuda([ - "@local_config_cuda//cuda:cudnn", + "@local_config_cuda//cuda:cudnn_header", ]), ) diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl index 2a37c65bc7..f6b497f813 100644 --- a/third_party/gpus/cuda/BUILD.tpl +++ b/third_party/gpus/cuda/BUILD.tpl @@ -128,6 +128,15 @@ 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}"], |