未验证 提交 8372329b 编写于 作者: K kangguangli 提交者: GitHub

[NewExe] update run checks of standalone executor (#52313)

* update run checks of standalone executor

* remove mlu related code
上级 1da67779
...@@ -1585,8 +1585,6 @@ class Executor: ...@@ -1585,8 +1585,6 @@ class Executor:
program = pruned_program program = pruned_program
def _can_use_interpreter_core(program, place): def _can_use_interpreter_core(program, place):
if core.is_compiled_with_mlu():
return False
compiled = isinstance( compiled = isinstance(
program, compiler.CompiledProgram program, compiler.CompiledProgram
...@@ -1606,28 +1604,6 @@ class Executor: ...@@ -1606,28 +1604,6 @@ class Executor:
) )
return False return False
# Unsupported case 2: async mode
if (
compiled_program._build_strategy is not None
and compiled_program._build_strategy.async_mode
):
warnings.warn(
"Standalone executor is not used for async mode",
UserWarning,
)
return False
# Unsupported case 3: CUDA Graph
if (
compiled_program._build_strategy is not None
and compiled_program._build_strategy.allow_cuda_graph_capture
and not _is_cuda_graph_enable_standalone_executor()
):
warnings.warn(
"Standalone executor is not used for CUDA Graph when FLAGS_CUDA_GRAPH_USE_STANDALONE_EXECUTOR=0",
UserWarning,
)
return False
return True return True
if self._enable_interpreter_core and _can_use_interpreter_core( if self._enable_interpreter_core and _can_use_interpreter_core(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册