diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-15 08:53:08 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-15 08:57:14 -0800 |
commit | 2df1a5cffe483c4f43828eb5a1c499560056661e (patch) | |
tree | 78a5e2c4db724fe0b3b70354a680d7dc8965a7ad /tensorflow/stream_executor/host | |
parent | 483f51a259cd31294f80229dee901919fa9ab698 (diff) |
Rename StreamExecutorInterface::BlockHostUntilDoneWithStatus to BlockHostUntilDone.
PiperOrigin-RevId: 179198370
Diffstat (limited to 'tensorflow/stream_executor/host')
-rw-r--r-- | tensorflow/stream_executor/host/host_gpu_executor.cc | 4 | ||||
-rw-r--r-- | tensorflow/stream_executor/host/host_gpu_executor.h | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/stream_executor/host/host_gpu_executor.cc b/tensorflow/stream_executor/host/host_gpu_executor.cc index d103dcd033..542f521ef7 100644 --- a/tensorflow/stream_executor/host/host_gpu_executor.cc +++ b/tensorflow/stream_executor/host/host_gpu_executor.cc @@ -162,7 +162,7 @@ void HostExecutor::DeallocateStream(Stream *stream) {} bool HostExecutor::CreateStreamDependency(Stream *dependent, Stream *other) { AsHostStream(dependent)->EnqueueTask( - [other]() { SE_CHECK_OK(other->BlockHostUntilDoneWithStatus()); }); + [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); AsHostStream(dependent)->BlockUntilDone(); return true; } @@ -177,7 +177,7 @@ bool HostExecutor::StopTimer(Stream *stream, Timer *timer) { return true; } -port::Status HostExecutor::BlockHostUntilDoneWithStatus(Stream *stream) { +port::Status HostExecutor::BlockHostUntilDone(Stream *stream) { AsHostStream(stream)->BlockUntilDone(); return port::Status::OK(); } diff --git a/tensorflow/stream_executor/host/host_gpu_executor.h b/tensorflow/stream_executor/host/host_gpu_executor.h index e884554a15..e2c0e6d6b7 100644 --- a/tensorflow/stream_executor/host/host_gpu_executor.h +++ b/tensorflow/stream_executor/host/host_gpu_executor.h @@ -139,7 +139,7 @@ class HostExecutor : public internal::StreamExecutorInterface { bool StopTimer(Stream *stream, Timer *timer) override; - port::Status BlockHostUntilDoneWithStatus(Stream *stream) override; + port::Status BlockHostUntilDone(Stream *stream) override; int PlatformDeviceCount() override { return 1; } |