diff options
| author |  avijit-nervana <avijit.chakraborty@intel.com> | 2018-08-06 13:34:24 -0700 |
|---|---|---|
| committer | avijit-nervana <avijit.chakraborty@intel.com> | 2018-08-06 13:34:24 -0700 |
| commit | 1149ad359f4e51a4e2c37a1dab8112056a38ef9b (patch) | |
| tree | 5b5e99bc5e3c6d6859620d45a595bbaaec9d0f9d /tensorflow/tensorflow.bzl | |
| parent | 46c2eafd65fd55d0837a9a86e8843f7f6d615990 (diff) | |
| parent | 9e35139d16bf259794a23e60c9f2b3f4e38c3b48 (diff) | |
Merge remote-tracking branch 'upstream/master'
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 45c8117304..443c582360 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -1102,6 +1102,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, |