diff options
| author |  Peter Hawkins <phawkins@google.com> | 2016-12-05 06:27:11 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-05 06:44:09 -0800 |
| commit | e4a4e922ed4be890f4273a7a276768dfefe80a4c (patch) | |
| tree | b724c791eb601946961899a621ea4335babfd1e0 /tensorflow/stream_executor/BUILD | |
| parent | b00e490c3834d72aecce445f72b3a717b068d9db (diff) | |
Add a Host platform to the stream executor library that runs code on the CPU.
Avoid using a linker-initialized mutex in multi_platform_manager to work around a initialization/destruction ordering problem on Mac.
Include the CUDA platform ID module even in non-CUDA builds.
Change: 141045549
Diffstat (limited to 'tensorflow/stream_executor/BUILD')
| -rw-r--r-- | tensorflow/stream_executor/BUILD | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD index 256b128750..a84f9d5f9d 100644 --- a/tensorflow/stream_executor/BUILD +++ b/tensorflow/stream_executor/BUILD @@ -7,6 +7,8 @@ cc_library( srcs = glob( [ "*.cc", + "host/*.cc", + "cuda/cuda_platform_id.cc", "lib/*.cc", "platform/default/*.cc", ], @@ -14,13 +16,17 @@ cc_library( "**/*_test.cc", ], ) + if_cuda( - glob([ - "cuda/*.cc", - ]), + glob( + [ + "cuda/*.cc", + ], + exclude = ["cuda/cuda_platform_id.cc"], + ), ), hdrs = glob([ "*.h", "cuda/*.h", + "host/*.h", "lib/*.h", "lib/gtl/*.h", "platform/**/*.h", |