From 16caedcc746867e26258fde0a018445e1423c5ca Mon Sep 17 00:00:00 2001 From: Bairen Yi Date: Sun, 12 May 2019 21:32:37 +0800 Subject: [PATCH] verbs: fixes GPU build broken by 4c3a1fb Use the default argument as in DeviceContext::CopyCPUTensorToDevice. Signed-off-by: Bairen Yi --- tensorflow/contrib/verbs/rdma.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/contrib/verbs/rdma.cc b/tensorflow/contrib/verbs/rdma.cc index b4b6b705f4b..121b483e7d1 100644 --- a/tensorflow/contrib/verbs/rdma.cc +++ b/tensorflow/contrib/verbs/rdma.cc @@ -1634,7 +1634,7 @@ void RdmaTensorRequest::RecvTensorContent() { [this](const Status& s) { CHECK(s.ok()) << "copy tensor to gpu sync"; Done(s); - }); + }, true /*sync_dst_compute*/); return; } #endif