diff --git a/python/paddle/fluid/compiler.py b/python/paddle/fluid/compiler.py
index b24cec044f1918ea9962dc10fc3939444bb340e8..403ceda87b137c7c11d02a9e9dce7fff8c3f0361 100644
--- a/python/paddle/fluid/compiler.py
+++ b/python/paddle/fluid/compiler.py
@@ -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(