mp test to serial test

PaddlePaddle · Dec 19, 2022 · e826914 · e826914
1 parent 3c67428
commit e826914
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 63 deletions.
diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt b/python/paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt
@@ -60,9 +60,10 @@ if(WITH_DISTRIBUTE AND WITH_GPU)
   py_test_modules(test_pass_amp MODULES test_pass_amp ENVS ${dist_ENVS})
   set_tests_properties(test_pass_amp PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE"
                                                 TIMEOUT 50)
-  py_test_modules(test_pass_bf16 MODULES test_pass_bf16 ENVS ${dist_ENVS})
-  set_tests_properties(test_pass_bf16 PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE"
-                                                 TIMEOUT 50)
+  py_test_modules(bf16_pass_unittest MODULES bf16_pass_unittest ENVS
+                  ${dist_ENVS})
+  set_tests_properties(bf16_pass_unittest
+                       PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 50)
   py_test_modules(test_engine_callbacks MODULES test_engine_callbacks)
   set_tests_properties(test_engine_callbacks
                        PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 50)
@@ -79,6 +80,7 @@ if(WITH_DISTRIBUTE AND WITH_GPU)
   set_tests_properties(test_selective_recompute PROPERTIES TIMEOUT 50)
   py_test_modules(test_tuning_recompute MODULES test_tuning_recompute)
   set_tests_properties(test_tuning_recompute PROPERTIES TIMEOUT 240)
+
   py_test_modules(test_while_op_completion MODULES test_while_op_completion
                   ENVS ${dist_ENVS})
   py_test_modules(test_while_op_partition MODULES test_while_op_partition ENVS

diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/bf16_pass_unittest.py b/python/paddle/fluid/tests/unittests/auto_parallel/bf16_pass_unittest.py
@@ -77,7 +77,7 @@ def get_engine(self, use_bf16=False):
         strategy = apply_pass(use_bf16)
         clip = paddle.nn.ClipGradByGlobalNorm(self.clip_norm)
         opt = paddle.optimizer.AdamW(learning_rate=0.00001, grad_clip=clip)
-        model, loss = generate_model("mp")
+        model, loss = generate_model("serial")
 
         engine = auto.Engine(model, loss, opt, strategy=strategy)
         self.init(engine)

diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/test_pass_bf16.py b/python/paddle/fluid/tests/unittests/auto_parallel/test_pass_bf16.py