diff options
| author |  Smit Hinsu <hinsu@google.com> | 2018-03-28 18:26:46 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-28 18:31:16 -0700 |
| commit | 59a12553545c3d8f957a1a6e618561d4228f7f59 (patch) | |
| tree | 409c242998a1415223f52ccbabfcf1155056efa3 /tensorflow/stream_executor/BUILD | |
| parent | 17dfe3ed7db7fb4d41f8933adead4737c30a92c9 (diff) | |
Relax CuDNN version requirements because CuDNN is backwards compatible within
a major release starting with CuDNN 7.0
PiperOrigin-RevId: 190869028
Diffstat (limited to 'tensorflow/stream_executor/BUILD')
| -rw-r--r-- | tensorflow/stream_executor/BUILD | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD index 1865240014..27cdb860fe 100644 --- a/tensorflow/stream_executor/BUILD +++ b/tensorflow/stream_executor/BUILD @@ -56,7 +56,10 @@ cc_library( [ "cuda/*.cc", ], - exclude = ["cuda/cuda_platform_id.cc"], + exclude = [ + "cuda/*_test.cc", + "cuda/cuda_platform_id.cc", + ], ), ), copts = select({ @@ -72,6 +75,7 @@ cc_library( ":stream_executor", "//tensorflow/core:lib", "//tensorflow/core/kernels:ops_util", + "@com_google_absl//absl/strings", "@local_config_cuda//cuda:cuda_headers", ] + if_cuda_is_configured([ "//tensorflow/core:cuda", |