oproj TP support acl graph (#4073)

### What this PR does / why we need it?
Reference #2167 and orpoj TP supports ACL graph.

- vLLM version: v0.11.0
- vLLM main:
83f478bb19

---------

Signed-off-by: zzhx1 <zzh_201018@outlook.com>
This commit is contained in:
zzhxxx
2025-11-11 19:39:06 +08:00
committed by GitHub
parent 0e6e08e939
commit 46a41b26d3
2 changed files with 4 additions and 1 deletions

View File

@@ -359,4 +359,7 @@ class TestAscendConfig(TestBase):
}
test_vllm_config.parallel_config = ParallelConfig(
data_parallel_size=4, tensor_parallel_size=1)
model_path = os.path.join(os.path.dirname(__file__), "fake_weight")
test_vllm_config.model_config = ModelConfig(model=model_path,
enforce_eager=True)
init_ascend_config(test_vllm_config)

View File

@@ -92,7 +92,7 @@ class AscendConfig:
raise AssertionError(
"oproj_tensor_parallel_size is only supported in the pure DP scenario"
)
if not self.torchair_graph_config.enabled:
if vllm_config.model_config.enforce_eager is True:
raise AssertionError(
"oproj_tensor_parallel_size is only supported in graph mode"
)