diff options
author | Jonathan Shen <jonathanasdf@google.com> | 2018-08-02 15:26:37 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-02 15:31:23 -0700 |
commit | f8afea0e1dc7226d2d4e6bc9fb75ba7094fa727e (patch) | |
tree | 3e7588fa3d0e0675e3976b51fef8c03f86320b95 /tensorflow/tensorflow.bzl | |
parent | a12135ff5f8cb786608c1caf3460efd61f32c4c5 (diff) |
Internal change.
PiperOrigin-RevId: 207176147
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 4 |
1 files changed, 4 insertions, 0 deletions
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, |