aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/host
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-12-15 08:53:08 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-12-15 08:57:14 -0800
commit2df1a5cffe483c4f43828eb5a1c499560056661e (patch)
tree78a5e2c4db724fe0b3b70354a680d7dc8965a7ad /tensorflow/stream_executor/host
parent483f51a259cd31294f80229dee901919fa9ab698 (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.cc4
-rw-r--r--tensorflow/stream_executor/host/host_gpu_executor.h2
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; }