提交 e2ace032 编写于 作者: Q qiaolongfei

rename RunAsyncUpdate to RunAsyncLoop

上级 f997c9b7
...@@ -199,9 +199,10 @@ void ListenAndServOp::RunSyncLoop(framework::Executor *executor, ...@@ -199,9 +199,10 @@ void ListenAndServOp::RunSyncLoop(framework::Executor *executor,
} // while(true) } // while(true)
} }
void ListenAndServOp::RunAsyncUpdate( void ListenAndServOp::RunAsyncLoop(framework::Executor *executor,
framework::Executor *executor, framework::ProgramDesc *program, framework::ProgramDesc *program,
framework::Scope *recv_scope, framework::BlockDesc *prefetch_block) const { framework::Scope *recv_scope,
framework::BlockDesc *prefetch_block) const {
// grad name to block id // grad name to block id
std::unordered_map<std::string, int32_t> grad_to_id; std::unordered_map<std::string, int32_t> grad_to_id;
std::unordered_map<int32_t, std::string> id_to_grad; std::unordered_map<int32_t, std::string> id_to_grad;
...@@ -305,7 +306,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, ...@@ -305,7 +306,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
if (sync_mode) { if (sync_mode) {
RunSyncLoop(&executor, program, &recv_scope, prefetch_block); RunSyncLoop(&executor, program, &recv_scope, prefetch_block);
} else { } else {
RunAsyncUpdate(&executor, program, &recv_scope, prefetch_block); RunAsyncLoop(&executor, program, &recv_scope, prefetch_block);
} }
} }
......
...@@ -46,10 +46,10 @@ class ListenAndServOp : public framework::OperatorBase { ...@@ -46,10 +46,10 @@ class ListenAndServOp : public framework::OperatorBase {
framework::Scope* recv_scope, framework::Scope* recv_scope,
framework::BlockDesc* prefetch_block) const; framework::BlockDesc* prefetch_block) const;
void RunAsyncUpdate(framework::Executor* executor, void RunAsyncLoop(framework::Executor* executor,
framework::ProgramDesc* program, framework::ProgramDesc* program,
framework::Scope* recv_scope, framework::Scope* recv_scope,
framework::BlockDesc* prefetch_block) const; framework::BlockDesc* prefetch_block) const;
void Stop() override; void Stop() override;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册