From 7fd0d24e0cf185251d861a81eabcda3a37b907fa Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Mon, 19 Mar 2018 20:13:35 +0800
Subject: [PATCH] Add lgo

---
 paddle/fluid/framework/parallel_executor.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc
index bc9035b302d..df04cfc4619 100644
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@@ -149,9 +149,15 @@ struct ScaleLossGradOpHandle : public OpHandle {
       auto stream =
           static_cast<platform::CUDADeviceContext *>(this->dev_ctx_[place_])
               ->stream();
+      VLOG(3) << "1";
+      PADDLE_ENFORCE(cudaGetLastError());
+      VLOG(3) << "2";
       memory::Copy(boost::get<platform::CUDAPlace>(place_), tmp,
                    platform::CPUPlace(), &coeff_, sizeof(float), stream);
+      PADDLE_ENFORCE(cudaGetLastError());
+      VLOG(3) << "3";
       PADDLE_ENFORCE(cudaEventRecord(ev_, stream));
+      VLOG(3) << "4";
     }
   }
 
-- 
GitLab