PaddlePaddle
diff --git a/‎test/deprecated/legacy_test/CMakeLists.txt
Copy file name to clipboard
+3-11Lines changed: 3 additions & 11 deletions b/‎test/deprecated/legacy_test/CMakeLists.txt
Copy file name to clipboard
+3-11Lines changed: 3 additions & 11 deletions
diff --git a/‎test/deprecated/legacy_test/test_dist_sparse_tensor_load_adam.py
Copy file name to clipboard
-47Lines changed: 0 additions & 47 deletions b/‎test/deprecated/legacy_test/test_dist_sparse_tensor_load_adam.py
Copy file name to clipboard
-47Lines changed: 0 additions & 47 deletions
diff --git a/‎test/legacy_test/CMakeLists.txt
Copy file name to clipboard
+6Lines changed: 6 additions & 0 deletions b/‎test/legacy_test/CMakeLists.txt
Copy file name to clipboard
+6Lines changed: 6 additions & 0 deletions
diff --git a/‎test/deprecated/legacy_test/test_adaptive_max_pool3d.py renamed to ‎test/legacy_test/test_adaptive_max_pool3d.py
Copy file name to clipboard b/‎test/deprecated/legacy_test/test_adaptive_max_pool3d.py renamed to ‎test/legacy_test/test_adaptive_max_pool3d.py
Copy file name to clipboard
diff --git a/‎test/deprecated/legacy_test/test_arg_min_max_op.py renamed to ‎test/legacy_test/test_arg_min_max_op.py
Copy file name to clipboard
+1-1Lines changed: 1 addition & 1 deletion b/‎test/deprecated/legacy_test/test_arg_min_max_op.py renamed to ‎test/legacy_test/test_arg_min_max_op.py
Copy file name to clipboard
+1-1Lines changed: 1 addition & 1 deletion
diff --git a/‎test/deprecated/legacy_test/test_bitwise_shift_op.py renamed to ‎test/legacy_test/test_bitwise_shift_op.py
Copy file name to clipboard b/‎test/deprecated/legacy_test/test_bitwise_shift_op.py renamed to ‎test/legacy_test/test_bitwise_shift_op.py
Copy file name to clipboard
diff --git a/‎test/deprecated/legacy_test/test_clip_grad_norm_.py renamed to ‎test/legacy_test/test_clip_grad_norm_.py
Copy file name to clipboard b/‎test/deprecated/legacy_test/test_clip_grad_norm_.py renamed to ‎test/legacy_test/test_clip_grad_norm_.py
Copy file name to clipboard
diff --git a/‎test/deprecated/legacy_test/test_dist_fleet_trainer_desc_config.py renamed to ‎test/legacy_test/test_dist_fleet_trainer_desc_config.py
Copy file name to clipboard b/‎test/deprecated/legacy_test/test_dist_fleet_trainer_desc_config.py renamed to ‎test/legacy_test/test_dist_fleet_trainer_desc_config.py
Copy file name to clipboard
diff --git a/‎test/legacy_test/test_dist_sparse_tensor_load_adam.py
Copy file name to clipboard
+86Lines changed: 86 additions & 0 deletions b/‎test/legacy_test/test_dist_sparse_tensor_load_adam.py
Copy file name to clipboard
+86Lines changed: 86 additions & 0 deletions
diff --git a/‎test/deprecated/legacy_test/test_dist_tree_index.py renamed to ‎test/legacy_test/test_dist_tree_index.py
Copy file name to clipboard b/‎test/deprecated/legacy_test/test_dist_tree_index.py renamed to ‎test/legacy_test/test_dist_tree_index.py
Copy file name to clipboard
@@ -439,7 +439,7 @@ endif()
 
 # Some ops need to check results when gc is enabled
 # Currently, only ops that register NoNeedBufferVarsInference need to do this test
-set(TEST_OPS_WITH_GC test_slice_op test_slice_op_deprecated)
+set(TEST_OPS_WITH_GC test_slice_op_deprecated)
 
 foreach(TEST_OP ${TEST_OPS_WITH_GC})
   list(REMOVE_ITEM TEST_OPS ${TEST_OP})
@@ -637,7 +637,6 @@ set_tests_properties(test_decoupled_py_reader_deprecated PROPERTIES TIMEOUT 120)
 set_tests_properties(test_fuse_bn_act_pass_deprecated PROPERTIES TIMEOUT 120)
 set_tests_properties(test_conv2d_api_deprecated PROPERTIES TIMEOUT 120)
 set_tests_properties(test_regularizer_deprecated PROPERTIES TIMEOUT 150)
-set_tests_properties(test_slice_op PROPERTIES TIMEOUT 120)
 set_tests_properties(test_slice_op_deprecated PROPERTIES TIMEOUT 120)
 set_tests_properties(test_dataloader_keep_order_deprecated PROPERTIES TIMEOUT
                                                                       120)
@@ -658,14 +657,8 @@ endif()
 set_tests_properties(test_inplace_addto_strategy_deprecated PROPERTIES TIMEOUT
                                                                        120)
 
-set(TEST_CINN_OPS
-    test_softmax_op
-    test_slice_op
-    test_slice_op_deprecated
-    test_layer_norm_op_deprecated
-    test_where_op
-    test_arg_min_max_op
-    test_instance_norm_op_deprecated)
+set(TEST_CINN_OPS test_slice_op_deprecated test_layer_norm_op_deprecated
+                  test_where_op test_instance_norm_op_deprecated)
 
 foreach(TEST_CINN_OP ${TEST_CINN_OPS})
   if(WITH_CINN)
@@ -685,7 +678,6 @@ set_tests_properties(test_conditional_block_deprecated
 
 # These UTs are to temporarily test static build for standalone_executor, will be removed after static build is enabled by default.
 set(STATIC_BUILD_TESTS
-    test_arg_min_max_op
     test_batch_norm_op_deprecated
     test_decoupled_py_reader_deprecated
     test_fuse_bn_act_pass_deprecated
 
@@ -502,6 +502,7 @@ endif()
 # Currently, only ops that register NoNeedBufferVarsInference need to do this test
 set(TEST_OPS_WITH_GC
     test_affine_channel_op
+    test_slice_op
     test_scatter_op
     test_concat_op
     test_elementwise_add_op
@@ -1040,6 +1041,9 @@ set(TEST_CINN_OPS
     test_pad_op
     test_pool2d_op
     test_erf_op
+    test_arg_min_max_op
+    test_slice_op
+    test_softmax_op
     test_cumsum_op)
 
 foreach(TEST_CINN_OP ${TEST_CINN_OPS})
@@ -1066,6 +1070,7 @@ set_tests_properties(
 
 # These UTs are to temporarily test static build for standalone_executor, will be removed after static build is enabled by default.
 set(STATIC_BUILD_TESTS
+    test_arg_min_max_op
     test_adagrad_op
     test_reduce_op
     test_adamw_op
@@ -1257,3 +1262,4 @@ set_tests_properties(test_inplace_softmax_with_cross_entropy PROPERTIES TIMEOUT
                                                                         120)
 set_tests_properties(test_imperative_selected_rows_to_lod_tensor
                      PROPERTIES TIMEOUT 200)
+set_tests_properties(test_slice_op PROPERTIES TIMEOUT 120)
@@ -20,7 +20,7 @@
 
 import numpy as np
 from op_test import OpTest, convert_float_to_uint16
-from test_attribute_var_deprecated import UnittestBase
+from test_attribute_var import UnittestBase
 
 import paddle
 from paddle.base import Program, program_guard
 
@@ -0,0 +1,86 @@
+#   Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+
+os.environ['FLAGS_enable_pir_api'] = '0'
+
+import os
+import unittest
+
+import paddle
+from paddle import base
+from paddle.distributed.fleet import fleet
+from paddle.distributed.fleet.base import role_maker
+
+
+class TestSparseLoadProgram(unittest.TestCase):
+    """
+    Test Sparse load operator.
+    """
+
+    def setUp(self):
+        os.environ["PADDLE_PSERVERS_IP_PORT_LIST"] = (
+            "127.0.0.1:4001,127.0.0.1:4002"
+        )
+        os.environ["PADDLE_TRAINERS_NUM"] = str(2)
+        os.environ["TRAINING_ROLE"] = "PSERVER"
+        os.environ["PADDLE_PORT"] = "4001"
+        os.environ["POD_IP"] = "127.0.0.1"
+        role = role_maker.PaddleCloudRoleMaker()
+        fleet.init(role)
+        self.strategy = paddle.distributed.fleet.DistributedStrategy()
+        self.strategy.a_sync = True
+
+    def net(self):
+        train_program = base.Program()
+        startup_program = base.Program()
+        scope = base.Scope()
+        with base.scope_guard(scope):
+            with base.program_guard(train_program, startup_program):
+                with base.unique_name.guard():
+                    inputs = paddle.static.data(
+                        'input', shape=[None, 1], dtype="int64"
+                    )
+                    emb = paddle.static.nn.embedding(
+                        inputs, is_sparse=True, size=[10000, 128]
+                    )
+                    fc1 = paddle.static.nn.fc(
+                        x=emb, size=128, activation="relu"
+                    )
+                    fc2 = paddle.static.nn.fc(x=fc1, size=64, activation="relu")
+                    loss = paddle.mean(fc2)
+            return scope, train_program, startup_program, loss
+
+
+class TestSparseLoadProgramAdam(TestSparseLoadProgram):
+    """
+    Test Sparse load operator.
+    """
+
+    def test_server_init(self):
+        scope, train_program, startup_program, loss = self.net()
+        with base.scope_guard(scope):
+            with base.program_guard(train_program, startup_program):
+                optimizer = paddle.optimizer.Adam(1e-3)
+                optimizer = fleet.distributed_optimizer(
+                    optimizer, self.strategy
+                )
+                optimizer.minimize(loss)
+                fleet.init_server()
+
+
+if __name__ == "__main__":
+    paddle.enable_static()
+    unittest.main()