diff --git a/paddle/fluid/operators/listen_and_serv_op.cc b/paddle/fluid/operators/listen_and_serv_op.cc index 7e2f8e74d673396927eb20e53b0e96ff72a595f2..e5e447164f52b9bae149244ec4c1af48dc48e8d6 100644 --- a/paddle/fluid/operators/listen_and_serv_op.cc +++ b/paddle/fluid/operators/listen_and_serv_op.cc @@ -199,9 +199,10 @@ void ListenAndServOp::RunSyncLoop(framework::Executor *executor, } // while(true) } -void ListenAndServOp::RunAsyncUpdate( - framework::Executor *executor, framework::ProgramDesc *program, - framework::Scope *recv_scope, framework::BlockDesc *prefetch_block) const { +void ListenAndServOp::RunAsyncLoop(framework::Executor *executor, + framework::ProgramDesc *program, + framework::Scope *recv_scope, + framework::BlockDesc *prefetch_block) const { // grad name to block id std::unordered_map grad_to_id; std::unordered_map id_to_grad; @@ -305,7 +306,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, if (sync_mode) { RunSyncLoop(&executor, program, &recv_scope, prefetch_block); } else { - RunAsyncUpdate(&executor, program, &recv_scope, prefetch_block); + RunAsyncLoop(&executor, program, &recv_scope, prefetch_block); } } diff --git a/paddle/fluid/operators/listen_and_serv_op.h b/paddle/fluid/operators/listen_and_serv_op.h index 2372e3cf3ee20abdfeb32655c9f58e9d579937cd..3cc0f3047733bea94daa310cd39cb0a4f44bef85 100644 --- a/paddle/fluid/operators/listen_and_serv_op.h +++ b/paddle/fluid/operators/listen_and_serv_op.h @@ -46,10 +46,10 @@ class ListenAndServOp : public framework::OperatorBase { framework::Scope* recv_scope, framework::BlockDesc* prefetch_block) const; - void RunAsyncUpdate(framework::Executor* executor, - framework::ProgramDesc* program, - framework::Scope* recv_scope, - framework::BlockDesc* prefetch_block) const; + void RunAsyncLoop(framework::Executor* executor, + framework::ProgramDesc* program, + framework::Scope* recv_scope, + framework::BlockDesc* prefetch_block) const; void Stop() override;