renae

7 years ago · f57978e6b5
parent fa3d47009b
commit f57978e6b5
2 changed files with 5 additions and 6 deletions
--- a/paddle/fluid/operators/distributed/grpc_client.h
+++ b/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;

--- a/paddle/fluid/operators/distributed/rpc_client.h
+++ b/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