From 4349f663375ecbb7e678d1e86606380e42d431ae Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 1 Jun 2018 07:30:28 -0700
Subject: Resubmitting CL 196349902: Adding cuDNN header dependency to targets
 that include the cuDNN header file.

PiperOrigin-RevId: 198869605
---
 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