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

verbs: fixes GPU build broken by 4c3a1fb

Browse Source 
Use the default argument as in DeviceContext::CopyCPUTensorToDevice.

Signed-off-by: Bairen Yi <byi@connect.ust.hk>
This commit is contained in:
Bairen Yi 2019-05-12 21:32:37 +08:00 committed by GitHub
parent 7d27d511c1
commit 16caedcc74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1634,7 +1634,7 @@ void RdmaTensorRequest::RecvTensorContent() {
[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