From e8d3745c0f8e2c0c891b86172e81b937d122f21b Mon Sep 17 00:00:00 2001 From: gongweibao Date: Thu, 19 Sep 2019 13:57:30 +0800 Subject: [PATCH] change _origin_program test=develop (#19863) change _origin_program test=develop --- python/paddle/fluid/incubate/fleet/collective/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/incubate/fleet/collective/__init__.py b/python/paddle/fluid/incubate/fleet/collective/__init__.py index 26b2dbeb29c..32a45c2dc9e 100644 --- a/python/paddle/fluid/incubate/fleet/collective/__init__.py +++ b/python/paddle/fluid/incubate/fleet/collective/__init__.py @@ -45,6 +45,7 @@ class Collective(Fleet): self.startup_program = None self._origin_program = None + self._transpiled_program = None self.main_program = None def init_worker(self): @@ -352,7 +353,8 @@ class CollectiveOptimizer(DistributedOptimizer): parameter_list=parameter_list, no_grad_set=no_grad_set) - fleet._origin_program = main_program + fleet._origin_program = main_program.clone(for_test=False) + fleet._transpiled_program = main_program fleet.main_program = self._try_to_compile(startup_program, main_program) return optimize_ops, param_grads -- GitLab