bug fix (#43153)

python/paddle/distributed/auto_parallel/completion.py

@@ -1307,7 +1307,7 @@ class Completer:
         if serial_main_program is None:
             serial_main_program = self._dist_context.serial_main_program
         else:
-            self._dist_context.serial_main_program = serial_main_program
+            self._dist_context._serial_main_program = serial_main_program
 
         import time
 

python/paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt

@@ -32,4 +32,5 @@ if(WITH_DISTRIBUTE AND WITH_GPU)
     py_test_modules(test_comm_cost MODULES test_comm_cost ENVS ${dist_ENVS})
     py_test_modules(test_comp_cost MODULES test_comp_cost ENVS ${dist_ENVS})
     py_test_modules(test_dist_context MODULES test_dist_context ENVS ${dist_ENVS})
+    py_test_modules(test_prim_dist_op MODULES test_prim_dist_op ENVS ${dist_ENVS})
 endif()