提交 932364a2 编写于 作者: Y Yu Yang

Sync dev

上级 dad7bdab
...@@ -155,7 +155,7 @@ struct ScaleLossGradOpHandle : public OpHandle { ...@@ -155,7 +155,7 @@ struct ScaleLossGradOpHandle : public OpHandle {
VLOG(3) << "2"; VLOG(3) << "2";
memory::Copy(boost::get<platform::CUDAPlace>(place_), tmp, memory::Copy(boost::get<platform::CUDAPlace>(place_), tmp,
platform::CPUPlace(), &coeff_, sizeof(float), stream); platform::CPUPlace(), &coeff_, sizeof(float), stream);
PADDLE_ENFORCE(cudaGetLastError()); PADDLE_ENFORCE(cudaDeviceSynchronize());
VLOG(3) << "3"; VLOG(3) << "3";
PADDLE_ENFORCE(cudaEventRecord(ev_, stream)); PADDLE_ENFORCE(cudaEventRecord(ev_, stream));
VLOG(3) << "4"; VLOG(3) << "4";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册