From 46a41b26d36926e79128a150294917a468e355e5 Mon Sep 17 00:00:00 2001 From: zzhxxx <96690582+zzhx1@users.noreply.github.com> Date: Tue, 11 Nov 2025 19:39:06 +0800 Subject: [PATCH] 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: https://github.com/vllm-project/vllm/commit/83f478bb19489b41e9d208b47b4bb5a95ac171ac --------- Signed-off-by: zzhx1 --- tests/ut/test_ascend_config.py | 3 +++ vllm_ascend/ascend_config.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/ut/test_ascend_config.py b/tests/ut/test_ascend_config.py index 4d3de7f4..718bc85f 100644 --- a/tests/ut/test_ascend_config.py +++ b/tests/ut/test_ascend_config.py @@ -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) diff --git a/vllm_ascend/ascend_config.py b/vllm_ascend/ascend_config.py index f947fc62..1fd1c67c 100644 --- a/vllm_ascend/ascend_config.py +++ b/vllm_ascend/ascend_config.py @@ -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" )