提交 4185dd48 编写于 作者: Y Yu Yang

Disable multi-thread

上级 631aa3d1
...@@ -84,8 +84,8 @@ struct OpHandle { ...@@ -84,8 +84,8 @@ struct OpHandle {
virtual ~OpHandle() {} virtual ~OpHandle() {}
virtual void Run() { PADDLE_THROW("Not implemented"); } virtual void Run() = 0;
virtual void Wait(platform::DeviceContext *waited_dev) {} virtual void Wait(platform::DeviceContext *waited_dev) = 0;
}; };
struct ComputationOpHandle : public OpHandle { struct ComputationOpHandle : public OpHandle {
...@@ -382,7 +382,6 @@ struct NCCLAllReduceOpHandle : public OpHandle { ...@@ -382,7 +382,6 @@ struct NCCLAllReduceOpHandle : public OpHandle {
} }
auto &var_name = static_cast<VarHandle *>(this->inputs_[0])->name_; auto &var_name = static_cast<VarHandle *>(this->inputs_[0])->name_;
VLOG(3) << "Invoke NCCL AllReduce";
int dtype = -1; int dtype = -1;
size_t numel = 0; size_t numel = 0;
...@@ -848,7 +847,8 @@ void ParallelExecutor::RunOp( ...@@ -848,7 +847,8 @@ void ParallelExecutor::RunOp(
LOG(FATAL) << "Unknown exception catched"; LOG(FATAL) << "Unknown exception catched";
} }
}; };
member_->pool_.enqueue(op_run); op_run();
// member_->pool_.enqueue(op_run);
} }
} // namespace framework } // namespace framework
} // namespace paddle } // namespace paddle
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册