From e53b6aba63a1635b137a57b15410f2eeda180e8e Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Tue, 20 Mar 2018 17:06:41 +0800 Subject: [PATCH] Use no thread --- paddle/fluid/framework/parallel_executor.cc | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc index e0b75b234..31a49575f 100644 --- a/paddle/fluid/framework/parallel_executor.cc +++ b/paddle/fluid/framework/parallel_executor.cc @@ -109,7 +109,6 @@ struct OpHandle { virtual void Wait(platform::DeviceContext *waited_dev) { if (platform::is_cpu_place(waited_dev->GetPlace()) || events_.empty()) { - VLOG(4) << "I am here"; for (auto &dev_ctx : dev_ctx_) { dev_ctx.second->Wait(); } @@ -255,7 +254,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 places_; @@ -397,8 +396,6 @@ struct NCCLAllReduceOpHandle : public OpHandle { } platform::dynload::ncclGroupEnd(); PADDLE_ENFORCE(cudaDeviceSynchronize()); - - VLOG(3) << "After NCCL"; } } }; -- GitLab