未验证 提交 6adeec4d 编写于 作者: Z zhaoyingli 提交者: GitHub

remove skip startup program when use newir (#57051)

上级 ad0e7a26
...@@ -833,14 +833,6 @@ class Engine: ...@@ -833,14 +833,6 @@ class Engine:
dist_main_program, self._place, dist_context 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: if self._executor is None:
self._executor = paddle.static.Executor(self._place) self._executor = paddle.static.Executor(self._place)
uninitialized = [] uninitialized = []
...@@ -868,11 +860,6 @@ class Engine: ...@@ -868,11 +860,6 @@ class Engine:
] ]
self._executor.run(dist_startup_prog) self._executor.run(dist_startup_prog)
if use_new_ir:
paddle.framework.set_flags(
{"FLAGS_enable_new_ir_in_executor": True}
)
def fit( def fit(
self, self,
train_data, train_data,
......
...@@ -2423,19 +2423,6 @@ def use_new_executor(): ...@@ -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): def get_pp_stage(dist_context, rank):
pp_idx = None pp_idx = None
for idx, process_mesh in enumerate(dist_context.process_meshes): for idx, process_mesh in enumerate(dist_context.process_meshes):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册