|
|
|
@ -178,9 +178,9 @@ class CompiledProgram(object):
|
|
|
|
|
# FIXME(dzhwinter): enable_inplace should be after memory_optimize
|
|
|
|
|
# if turn on python memory optimize, turn off the inplace_pass.
|
|
|
|
|
if self._build_strategy.memory_optimize is None:
|
|
|
|
|
self._build_strategy.memory_optimize = False if main._is_mem_optimized else True
|
|
|
|
|
self._build_strategy.memory_optimize = False if self._program._is_mem_optimized else True
|
|
|
|
|
if self._build_strategy.enable_inplace is None:
|
|
|
|
|
self._build_strategy.enable_inplace = False if main._is_mem_optimized else True
|
|
|
|
|
self._build_strategy.enable_inplace = False if self._program._is_mem_optimized else True
|
|
|
|
|
|
|
|
|
|
if self._build_strategy.num_trainers > 1 and trainers_endpoints:
|
|
|
|
|
assert self._build_strategy.num_trainers == len(
|
|
|
|
|