diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-11 23:35:29 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-11 23:38:57 -0800 |
commit | 383a3226a9ad08ac507a3fbd6c220c5c1e15a540 (patch) | |
tree | 23832854e4d84b9366ef57adfeb15121e525ac13 /tensorflow/stream_executor/host | |
parent | fcca16c5e3d8168ea95616956fdcf777798b164b (diff) |
Use BlockHostUntilDoneWithStatus in various places.
PiperOrigin-RevId: 178723711
Diffstat (limited to 'tensorflow/stream_executor/host')
-rw-r--r-- | tensorflow/stream_executor/host/host_gpu_executor.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/host/host_gpu_executor.cc b/tensorflow/stream_executor/host/host_gpu_executor.cc index 1fd8eeb881..d103dcd033 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]() { other->BlockHostUntilDone(); }); + [other]() { SE_CHECK_OK(other->BlockHostUntilDoneWithStatus()); }); AsHostStream(dependent)->BlockUntilDone(); return true; } |