From 1533bf12dfa057bc7e34be540a391cb83d4dc9bb Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 20 Mar 2018 18:38:02 +0800
Subject: [PATCH] Use event and single thread

---
 paddle/fluid/framework/parallel_executor.cc | 4 ++--
 paddle/fluid/memory/memory.cc               | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc
index daa19eb17c8..f1b8a20e41c 100644
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@@ -245,7 +245,7 @@ struct FetchOpHandle : public OpHandle {
 
 class ParallelExecutorPrivate {
  public:
-  explicit ParallelExecutorPrivate(size_t num_threads = 12)
+  explicit ParallelExecutorPrivate(size_t num_threads = 0)
       : pool_(num_threads == 0 ? nullptr : new ThreadPool(num_threads)) {}
 
   std::vector<platform::Place> places_;
@@ -669,7 +669,7 @@ void ParallelExecutor::BuildNCCLCommunicator() const {
 
 void ParallelExecutor::Run(const std::vector<std::string> &fetch_tensors,
                            const std::string &fetched_var_name) {
-  bool use_event = false;
+  bool use_event = true;
   auto fetched_data = std::make_shared<FetchedData>(fetch_tensors.size());
   // Version --> VarHandle
   member_->exception_.reset();
diff --git a/paddle/fluid/memory/memory.cc b/paddle/fluid/memory/memory.cc
index a12cdd45aa1..1985f1f4e68 100644
--- a/paddle/fluid/memory/memory.cc
+++ b/paddle/fluid/memory/memory.cc
@@ -90,7 +90,6 @@ size_t Used<platform::CUDAPlace>(platform::CUDAPlace place) {
 template <>
 void* Alloc<platform::CUDAPlace>(platform::CUDAPlace place, size_t size) {
   auto* buddy_allocator = GetGPUBuddyAllocator(place.device);
-  VLOG(30) << "Allocating " << size << " bytes on " << place;
   auto* ptr = buddy_allocator->Alloc(size);
   if (ptr == nullptr) {
     int cur_dev = platform::GetCurrentDeviceId();
-- 
GitLab