aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-07-12 10:57:48 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-12 11:01:37 -0700
commitf8ce8dc04f569b3bbd1ac7cc9e358a651530990d (patch)
treeb0c91b93e89ee1206e81c0556f891ce8412034f0 /tensorflow/stream_executor
parentf8044c89287b1d90510ceace4b53ec94abaffa50 (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.cc2
-rw-r--r--tensorflow/stream_executor/stream.cc6
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);