diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-01-26 15:55:26 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-26 16:07:43 -0800 |
| commit | 8845387fc6e8eb476bb42e620bd86b263bda41b9 (patch) | |
| tree | d5965c77f78b64f521489a7b5fd1df939b88affa /tensorflow/stream_executor/lib | |
| parent | b80fce2f901ac393cd7edcbe484221c95ff5ba90 (diff) | |
Change the way the host platform stream blocks so that it is usable with a condition variable implementation that doesn't support waiting with a predicate.
Remove unnecessary ToString calls in cuda_gpu_executor.cc.
Minor tweaks to header inclusions which should have no effect on code behavior.
Change: 145732882
Diffstat (limited to 'tensorflow/stream_executor/lib')
| -rw-r--r-- | tensorflow/stream_executor/lib/initialize.h | 1 | ||||
| -rw-r--r-- | tensorflow/stream_executor/lib/ptr_util.h | 2 |
2 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/lib/initialize.h b/tensorflow/stream_executor/lib/initialize.h index 1b8c03910c..9a09318a6c 100644 --- a/tensorflow/stream_executor/lib/initialize.h +++ b/tensorflow/stream_executor/lib/initialize.h @@ -19,6 +19,7 @@ limitations under the License. #include "tensorflow/stream_executor/platform/port.h" #if defined(PLATFORM_GOOGLE) +#include "tensorflow/stream_executor/platform/google/initialize.h" #else #undef REGISTER_MODULE_INITIALIZER diff --git a/tensorflow/stream_executor/lib/ptr_util.h b/tensorflow/stream_executor/lib/ptr_util.h index 2f06b8402c..3d5e56faf7 100644 --- a/tensorflow/stream_executor/lib/ptr_util.h +++ b/tensorflow/stream_executor/lib/ptr_util.h @@ -16,6 +16,8 @@ limitations under the License. #ifndef TENSORFLOW_STREAM_EXECUTOR_LIB_PTR_UTIL_H_ #define TENSORFLOW_STREAM_EXECUTOR_LIB_PTR_UTIL_H_ +#include <memory> + namespace perftools { namespace gputools { namespace port { |