From 7ba843e6b07693bb228d48e9701662492f33c29d Mon Sep 17 00:00:00 2001 From: zhaoyingli <86812880+zhaoyinglia@users.noreply.github.com> Date: Thu, 2 Jun 2022 10:06:43 +0800 Subject: [PATCH] bug fix (#43153) --- python/paddle/distributed/auto_parallel/completion.py | 2 +- .../paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/distributed/auto_parallel/completion.py b/python/paddle/distributed/auto_parallel/completion.py index 465c450c0b0..86632d81098 100644 --- a/python/paddle/distributed/auto_parallel/completion.py +++ b/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 diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt b/python/paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt index 381461130ed..8c8a6823105 100644 --- a/python/paddle/fluid/tests/unittests/auto_parallel/CMakeLists.txt +++ b/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() -- GitLab