From f8afea0e1dc7226d2d4e6bc9fb75ba7094fa727e Mon Sep 17 00:00:00 2001
From: Jonathan Shen <jonathanasdf@google.com>
Date: Thu, 2 Aug 2018 15:26:37 -0700
Subject: Internal change.

PiperOrigin-RevId: 207176147
---
 tensorflow/tensorflow.bzl | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'tensorflow/tensorflow.bzl')

diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 7d6a27b102..39db840884 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -1096,6 +1096,10 @@ def tf_kernel_library(
     tf_gpu_kernel_library(
         name=name + "_gpu", srcs=gpu_srcs, deps=deps, **kwargs)
     cuda_deps.extend([":" + name + "_gpu"])
+  kwargs["tags"] = kwargs.get("tags", []) + [
+      "req_dep=%s" % clean_dep("//tensorflow/core:gpu_lib"),
+      "req_dep=@local_config_cuda//cuda:cuda_headers",
+  ]
   tf_cuda_library(
       name=name,
       srcs=srcs,
-- 
cgit v1.2.3