diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-01 07:30:28 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-01 07:32:52 -0700 |
commit | 4349f663375ecbb7e678d1e86606380e42d431ae (patch) | |
tree | b18b412942fafcd6f3eb9e385683e6b6a9049c5e /third_party/gpus | |
parent | e6aca210f1082e4cb8cf3d0f775a79042b48f68a (diff) |
Resubmitting CL 196349902: Adding cuDNN header dependency to targets that include the cuDNN header file.
PiperOrigin-RevId: 198869605
Diffstat (limited to 'third_party/gpus')
-rw-r--r-- | third_party/gpus/cuda/BUILD.tpl | 9 |
1 files changed, 9 insertions, 0 deletions
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}"], |