renae

paddle/fluid/operators/distributed/grpc_client.h

@@ -211,9 +211,8 @@ class GRPCClient : public RPCClient {
   void AsyncSendFetchBarrier(const std::string& ep,
                              int64_t time_out = FLAGS_grpc_deadline) override;
 
-  void AsyncCheckpointNotify(
-      const std::string& ep, const std::string& dir,
-      int64_t time_out = RPCClient::rpc_time_out) override;
+  void AsyncCheckpointNotify(const std::string& ep, const std::string& dir,
+                             int64_t time_out = FLAGS_grpc_deadline) override;
 
   void Wait() override;
 

paddle/fluid/operators/distributed/rpc_client.h

@@ -56,9 +56,9 @@ class RPCClient {
   virtual void AsyncSendFetchBarrier(
       const std::string& ep, int64_t time_out = FLAGS_grpc_deadline) = 0;
 
-  virtual void AsyncCheckpointNotify(const std::string& ep,
-                                     const std::string& dir,
-                                     int64_t time_out = rpc_time_out) = 0;
+  virtual void AsyncCheckpointNotify(
+      const std::string& ep, const std::string& dir,
+      int64_t time_out = FLAGS_grpc_deadline) = 0;
 
   // SendComplete tells all the server that current trainer have no more data
   // to train, so that the pserver can reduce it's barrier count, and continue