diff options
author | 2017-03-22 12:41:11 -0800 | |
---|---|---|
committer | 2017-03-22 14:07:04 -0700 | |
commit | 9e5533fe3d04d4ce5b6f23b7362796d85b5fa5f2 (patch) | |
tree | 047e50c10f832de49cb35f6d6adcb95f859dc4f1 /tensorflow/tensorflow.bzl | |
parent | 5dff80f119d6b2dd70952b7d585a9a77666ea2f8 (diff) |
Fixing a bug with the tf_kernel_library rule where the exclude attribute in the globs didn't include the prefix. This causes issues with build rules such as contrib/rnn:gru_ops_kernels.
Change: 150921333
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 4edbd0b70e..471a2173aa 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -520,10 +520,11 @@ def tf_kernel_library(name, prefix=None, srcs=None, gpu_srcs=None, hdrs=None, if not gpu_srcs: gpu_srcs = [] gpu_srcs = gpu_srcs + native.glob([prefix + "*.cu.cc", prefix + "*.h"], - exclude = ["*test*"]) + exclude = [prefix + "*test*"]) srcs = srcs + native.glob([prefix + "*.cc"], - exclude = ["*test*", "*.cu.cc"]) - hdrs = hdrs + native.glob([prefix + "*.h"], exclude = ["*test*", "*.cu.h"]) + exclude = [prefix + "*test*", prefix + "*.cu.cc"]) + hdrs = hdrs + native.glob([prefix + "*.h"], exclude = [prefix + "*test*", + prefix + "*.cu.h"]) cuda_deps = ["//tensorflow/core:gpu_lib"] if gpu_srcs: |