diff --git a/paddle/fluid/operators/checkpoint_notify_op.cc b/paddle/fluid/operators/checkpoint_notify_op.cc index 31b725ec184ced83109f7d47851e9e962c555e02..e7a65b76a496c02549fc653fd3d49fc2c649ef73 100644 --- a/paddle/fluid/operators/checkpoint_notify_op.cc +++ b/paddle/fluid/operators/checkpoint_notify_op.cc @@ -39,8 +39,8 @@ class CheckpointNotifyOp : public framework::OperatorBase { std::string dir = Attr("dir"); std::string lookup_table_name = Attr("lookup_table"); - detail::RPCClient* rpc_client = - detail::RPCClient::GetInstance(); + distributed::RPCClient* rpc_client = + distributed::RPCClient::GetInstance(); for (size_t i = 0; i < epmap.size(); i++) { VLOG(3) << "checkpoint notify sending " << dir << " to " << epmap[i]; auto serial_looku_table = diff --git a/paddle/fluid/operators/listen_and_serv_op.cc b/paddle/fluid/operators/listen_and_serv_op.cc index 420c4e9e4ed2ef0b4a3b917f013360ad5a5dfa5b..df9cdae97df916b04071abd1ced26876732d2022 100644 --- a/paddle/fluid/operators/listen_and_serv_op.cc +++ b/paddle/fluid/operators/listen_and_serv_op.cc @@ -268,7 +268,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, request_get_handler_.get()); rpc_service_->RegisterRPC(distributed::kRequestPrefetch, request_prefetch_handler_.get()); - rpc_service_->RegisterRPC(detail::kRequestCheckpoint, + rpc_service_->RegisterRPC(distributed::kRequestCheckpoint, request_checkpoint_handler_.get()); auto *optimize_block = Attr(kOptimizeBlock);