From 932364a27597e141b167694d9ec94e615965cbfc Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Mon, 19 Mar 2018 20:21:50 +0800 Subject: [PATCH] Sync dev --- paddle/fluid/framework/parallel_executor.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc index c3a90149a..67e7078fb 100644 --- a/paddle/fluid/framework/parallel_executor.cc +++ b/paddle/fluid/framework/parallel_executor.cc @@ -155,7 +155,7 @@ struct ScaleLossGradOpHandle : public OpHandle { VLOG(3) << "2"; memory::Copy(boost::get(place_), tmp, platform::CPUPlace(), &coeff_, sizeof(float), stream); - PADDLE_ENFORCE(cudaGetLastError()); + PADDLE_ENFORCE(cudaDeviceSynchronize()); VLOG(3) << "3"; PADDLE_ENFORCE(cudaEventRecord(ev_, stream)); VLOG(3) << "4"; -- GitLab