remove skip startup program when use newir (#57051)

zhaoyinglia · web-flow · commit 6adeec4d2396 · 2023-09-08T10:46:53.000+08:00
diff --git a/python/paddle/distributed/auto_parallel/static/engine.py b/python/paddle/distributed/auto_parallel/static/engine.py
@@ -833,14 +833,6 @@ def _initialize(self, mode):
                 dist_main_program, self._place, dist_context
             )
 
-        # NOTE(zhaoyinglia): Skip startup program when use new ir temporarily.
-        use_new_ir = False
-        if auto_utils.use_new_ir():
-            use_new_ir = True
-            paddle.framework.set_flags(
-                {"FLAGS_enable_new_ir_in_executor": False}
-            )
-
         if self._executor is None:
             self._executor = paddle.static.Executor(self._place)
             uninitialized = []
@@ -868,11 +860,6 @@ def _initialize(self, mode):
             ]
             self._executor.run(dist_startup_prog)
 
-        if use_new_ir:
-            paddle.framework.set_flags(
-                {"FLAGS_enable_new_ir_in_executor": True}
-            )
-
     def fit(
         self,
         train_data,
diff --git a/python/paddle/distributed/auto_parallel/static/utils.py b/python/paddle/distributed/auto_parallel/static/utils.py
@@ -2423,19 +2423,6 @@ def use_new_executor():
     ]
 
 
-def use_new_ir():
-    enable_new_ir_in_executor = os.environ.get(
-        'FLAGS_enable_new_ir_in_executor', None
-    )
-    return enable_new_ir_in_executor in [
-        1,
-        '1',
-        True,
-        'True',
-        'true',
-    ]
-
-
 def get_pp_stage(dist_context, rank):
     pp_idx = None
     for idx, process_mesh in enumerate(dist_context.process_meshes):
