m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Sync dev

Browse Source
helinwang-patch-1
Yu Yang 7 years ago
parent dad7bdabd4
commit 932364a275
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
paddle/fluid/framework/parallel_executor.cc
Unescape View File

@ -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";

Write Preview
Loading…
Cancel
Save