File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -178,9 +178,9 @@ def _compile_data_parallel(self):
178
178
# FIXME(dzhwinter): enable_inplace should be after memory_optimize
179
179
# if turn on python memory optimize, turn off the inplace_pass.
180
180
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
182
182
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
184
184
185
185
if self ._build_strategy .num_trainers > 1 and trainers_endpoints :
186
186
assert self ._build_strategy .num_trainers == len (
You can’t perform that action at this time.
0 commit comments