diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-11 23:48:06 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-11 23:50:49 -0700 |
| commit | a436cf493d3a590572aec9fe574f0e9028e8b61e (patch) | |
| tree | 2a72dc460458b90a27f1075afbb01feee118ac59 /third_party/gpus | |
| parent | fc5250f97188e9b247845e32692d1c4ffad170c4 (diff) | |
Adding cuDNN header dependency to targets that include the cuDNN header file.
PiperOrigin-RevId: 196349902
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}"], |