change _origin_program test=develop (#19863)

change _origin_program test=develop
expand_as_op_1
gongweibao 5 years ago committed by GitHub
parent a7c440d303
commit e8d3745c0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -45,6 +45,7 @@ class Collective(Fleet):
self.startup_program = None self.startup_program = None
self._origin_program = None self._origin_program = None
self._transpiled_program = None
self.main_program = None self.main_program = None
def init_worker(self): def init_worker(self):
@ -352,7 +353,8 @@ class CollectiveOptimizer(DistributedOptimizer):
parameter_list=parameter_list, parameter_list=parameter_list,
no_grad_set=no_grad_set) 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) fleet.main_program = self._try_to_compile(startup_program, main_program)
return optimize_ops, param_grads return optimize_ops, param_grads

Loading…
Cancel
Save