From 7215454de6df78f4f9a49a99c5739f8bb360f5bc Mon Sep 17 00:00:00 2001 From: panchao-hub <315134829@qq.com> Date: Sat, 30 Aug 2025 15:49:48 +0800 Subject: [PATCH] bugfix for torchair graph (#2639) ### What this PR does / why we need it? bugfix for torchair graph ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? - vLLM version: v0.10.1.1 - vLLM main: https://github.com/vllm-project/vllm/commit/67c14906aaa480d4fee2606f31c784ae21f8a633 Signed-off-by: zhangdepeng Co-authored-by: zhangdepeng --- tests/ut/torchair/test_utils.py | 2 +- vllm_ascend/torchair/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ut/torchair/test_utils.py b/tests/ut/torchair/test_utils.py index ab85d28..8aebb9d 100644 --- a/tests/ut/torchair/test_utils.py +++ b/tests/ut/torchair/test_utils.py @@ -70,7 +70,7 @@ class TestTorchairUtils(TestBase): ), ("Qwen2ForCausalLM", "vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM"), - ("Qwen3ForCausalLM", + ("Qwen3MoeForCausalLM", "vllm_ascend.torchair.models.qwen3_moe:CustomQwen3MoeForCausalLM") ] diff --git a/vllm_ascend/torchair/utils.py b/vllm_ascend/torchair/utils.py index 8448ddc..2f3a28e 100644 --- a/vllm_ascend/torchair/utils.py +++ b/vllm_ascend/torchair/utils.py @@ -168,7 +168,7 @@ def register_torchair_model(): "vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM") ModelRegistry.register_model( - "Qwen3ForCausalLM", + "Qwen3MoeForCausalLM", "vllm_ascend.torchair.models.qwen3_moe:CustomQwen3MoeForCausalLM")