diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-12 10:57:48 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-12 11:01:37 -0700 |
commit | f8ce8dc04f569b3bbd1ac7cc9e358a651530990d (patch) | |
tree | b0c91b93e89ee1206e81c0556f891ce8412034f0 /tensorflow/stream_executor | |
parent | f8044c89287b1d90510ceace4b53ec94abaffa50 (diff) |
Automated rollback of commit f8044c89287b1d90510ceace4b53ec94abaffa50
PiperOrigin-RevId: 204327453
Diffstat (limited to 'tensorflow/stream_executor')
-rw-r--r-- | tensorflow/stream_executor/host/host_gpu_executor.cc | 2 | ||||
-rw-r--r-- | tensorflow/stream_executor/stream.cc | 6 |
2 files changed, 1 insertions, 7 deletions
diff --git a/tensorflow/stream_executor/host/host_gpu_executor.cc b/tensorflow/stream_executor/host/host_gpu_executor.cc index 8adf739b17..3cd97b3cf1 100644 --- a/tensorflow/stream_executor/host/host_gpu_executor.cc +++ b/tensorflow/stream_executor/host/host_gpu_executor.cc @@ -93,7 +93,7 @@ bool HostExecutor::MemcpyDeviceToDevice(Stream *stream, // the nature of the HostExecutor) memcpy on the stream (HostStream) // associated with the HostExecutor. AsHostStream(stream)->EnqueueTask( - [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); }); + [src_mem, dst_mem, size]() { memcpy(src_mem, dst_mem, size); }); return true; } diff --git a/tensorflow/stream_executor/stream.cc b/tensorflow/stream_executor/stream.cc index b3ed7e4452..9369183133 100644 --- a/tensorflow/stream_executor/stream.cc +++ b/tensorflow/stream_executor/stream.cc @@ -268,12 +268,6 @@ Stream::~Stream() { VLOG_CALL(); temporary_memory_manager_.ForceDeallocateAll(); - // Ensure the stream is completed. - auto status = BlockHostUntilDone(); - if (!status.ok()) { - LOG(WARNING) << "Error blocking host until done in stream destructor: " - << status; - } if (allocated_) { parent_->DeallocateStream(this); |