experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #28641 from byronyi:patch-4

Browse Source 
PiperOrigin-RevId: 248021303
This commit is contained in:
TensorFlower Gardener 2019-05-13 16:49:32 -07:00
commit b1c6a1accb
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/contrib/verbs/rdma.cc
View File

@ -1629,12 +1629,13 @@ void RdmaTensorRequest::RecvTensorContent() {
CountCopies(key_, (void*)DMAHelper::base(proxy_tensor_), CountCopies(key_, (void*)DMAHelper::base(proxy_tensor_),
(void*)DMAHelper::base(result_tensor_), (void*)DMAHelper::base(result_tensor_),
result_tensor_->TotalBytes(), false); result_tensor_->TotalBytes(), false);
GPUUtil::CopyCPUTensorToGPU(proxy_tensor_, recv_args_.device_context, GPUUtil::CopyCPUTensorToGPU(
dst_dev_, result_tensor_, proxy_tensor_, recv_args_.device_context, dst_dev_, result_tensor_,
[this](const Status& s) { [this](const Status& s) {
CHECK(s.ok()) << "copy tensor to gpu sync"; CHECK(s.ok()) << "copy tensor to gpu sync";
Done(s); Done(s);
}); },
true /*sync_dst_compute*/);
return; return;
} }
#endif #endif