diff --git a/paddle/fluid/distributed/service/ps_client.h b/paddle/fluid/distributed/service/ps_client.h
index 6c6eae8249041..3be83436cec34 100644
--- a/paddle/fluid/distributed/service/ps_client.h
+++ b/paddle/fluid/distributed/service/ps_client.h
@@ -66,7 +66,7 @@ class PSClient {
   PSClient(PSClient &&) = delete;
   PSClient(const PSClient &) = delete;
 
-  int32_t configure(
+  virtual int32_t configure(  // NOLINT
       const PSParameter &config,
       const std::map<uint64_t, std::vector<paddle::distributed::Region>>
           &regions,
@@ -114,9 +114,9 @@ class PSClient {
                                                 size_t region_num,
                                                 size_t table_id) = 0;
 
-  virtual std::future<int32_t> push_dense(const Region *regions,
-                                          size_t region_num,
-                                          size_t table_id) = 0;
+  //  virtual std::future<int32_t> push_dense(const Region *regions,
+  //                                          size_t region_num,
+  //                                          size_t table_id) = 0;
   // 使用keys进行pull请求，结果填充values
   // keys和values的个数均为num个，每个value占用select_size空间
   // future结束前keys和values缓冲区不能再次使用
@@ -222,10 +222,10 @@ class PSClient {
                                                  const uint64_t *keys,
                                                  const float **update_values,
                                                  size_t num, void *done) = 0;
-  virtual std::future<int32_t> push_sparse(size_t table_id,
-                                           const uint64_t *keys,
-                                           const float **update_values,
-                                           size_t num) = 0;
+  //  virtual std::future<int32_t> push_sparse(size_t table_id,
+  //                                           const uint64_t *keys,
+  //                                           const float **update_values,
+  //                                           size_t num) = 0;
 
  protected:
   virtual int32_t initialize() = 0;