From 3c9cea597e1e3075f8b56d0c7d11febe1a384033 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 20 Mar 2018 16:58:37 +0800
Subject: [PATCH] Add more log

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

diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc
index d1652a3030..24a9dcacf2 100644
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@@ -365,6 +365,7 @@ struct NCCLAllReduceOpHandle : public OpHandle {
         auto &p = static_cast<VarHandle *>(in)->place_;
         in->generated_op_->Wait(dev_ctx_[p]);
       }
+      VLOG(3) << "Before NCCL";
       PADDLE_ENFORCE(cudaDeviceSynchronize());
 
       auto &var_name = static_cast<VarHandle *>(this->inputs_[0])->name_;
@@ -394,8 +395,9 @@ struct NCCLAllReduceOpHandle : public OpHandle {
             nccl_ctx.comm, nccl_ctx.stream());
       }
       platform::dynload::ncclGroupEnd();
-
       PADDLE_ENFORCE(cudaDeviceSynchronize());
+
+      VLOG(3) << "After NCCL";
     }
   }
 };
-- 
GitLab