Sync dev

932364a2 · Yu Yang · dad7bdab · 932364a2
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/framework/parallel_executor.cc paddle/fluid/framework/parallel_executor.cc +1 -1

未找到文件。
--- 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<platform::CUDAPlace>(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";