[executor] feat: add supports_pp attr to executors (#21786)

Signed-off-by: Haibin Lin <haibin.lin@bytedance.com>
This commit is contained in:
H
2025-08-03 03:04:45 -07:00
committed by GitHub
parent 7de45db9a5
commit 24d1dffbeb
4 changed files with 17 additions and 8 deletions

View File

@@ -1490,14 +1490,18 @@ class EngineArgs:
and _warn_or_fallback("Engine in background thread")):
return False
if (self.pipeline_parallel_size > 1
and self.distributed_executor_backend
not in (ParallelConfig.distributed_executor_backend, "ray",
"mp", "external_launcher")):
name = "Pipeline Parallelism without Ray distributed executor " \
"or multiprocessing executor or external launcher"
_raise_or_fallback(feature_name=name, recommend_to_remove=False)
return False
if self.pipeline_parallel_size > 1:
supports_pp = getattr(self.distributed_executor_backend,
'supports_pp', False)
if not supports_pp and self.distributed_executor_backend not in (
ParallelConfig.distributed_executor_backend, "ray", "mp",
"external_launcher"):
name = "Pipeline Parallelism without Ray distributed " \
"executor or multiprocessing executor or external " \
"launcher"
_raise_or_fallback(feature_name=name,
recommend_to_remove=False)
return False
# The platform may be supported on V1, but off by default for now.
if not current_platform.default_v1( # noqa: SIM103