Skip to content

Commit b2eb623

Browse files
committed
cherry-picked changes. test=release/1.3
1 parent 7f4bdb1 commit b2eb623

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/compiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,9 @@ def _compile_data_parallel(self):
178178
# FIXME(dzhwinter): enable_inplace should be after memory_optimize
179179
# if turn on python memory optimize, turn off the inplace_pass.
180180
if self._build_strategy.memory_optimize is True:
181-
self._build_strategy.memory_optimize = False if main._is_mem_optimized else True
181+
self._build_strategy.memory_optimize = False if self._program._is_mem_optimized else True
182182
if self._build_strategy.enable_inplace is True:
183-
self._build_strategy.enable_inplace = False if main._is_mem_optimized else True
183+
self._build_strategy.enable_inplace = False if self._program._is_mem_optimized else True
184184

185185
if self._build_strategy.num_trainers > 1 and trainers_endpoints:
186186
assert self._build_strategy.num_trainers == len(

0 commit comments

Comments
 (0)