From a436cf493d3a590572aec9fe574f0e9028e8b61e Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 11 May 2018 23:48:06 -0700
Subject: Adding cuDNN header dependency to targets that include the cuDNN
 header file.

PiperOrigin-RevId: 196349902
---
 third_party/gpus/cuda/BUILD.tpl | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'third_party/gpus')

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
@@ -127,6 +127,15 @@ cc_library(
     visibility = ["//visibility:public"],
 )
 
+cc_library(
+    name = "cudnn_header",
+    includes = [
+        ".",
+        "cuda/include",
+    ],
+    visibility = ["//visibility:public"],
+)
+
 cc_library(
     name = "cufft",
     srcs = ["cuda/lib/%{cufft_lib}"],
-- 
cgit v1.2.3