[npu] Combine NPU test fixes from #35472 and #34516 (#36180)

This commit is contained in:
pllimax
2026-08-25 09:00:40 +08:00
committed by GitHub
parent 91e7e84ee5
commit d2d8ecea77
7 changed files with 92 additions and 28 deletions
@@ -20,7 +20,7 @@ ENVS = {
"HCCL_SOCKET_IFNAME": "lo",
"GLOO_SOCKET_IFNAME": "lo",
"HCCL_OP_EXPANSION_MODE": "AIV",
"HCCL_BUFFSIZE": "2000",
"DEEPEP_HCCL_BUFFSIZE": "2000",
}
OTHER_ARGS = [
@@ -54,7 +54,7 @@ OTHER_ARGS = [
]
class TestQwen3(TestNpuAccuracyTestCaseBase):
class TestQwen3_VL_30B_A3B_Thinking_MMMU(TestNpuAccuracyTestCaseBase):
model = QWEN3_VL_30B_A3B_THINKING_MODEL_PATH
envs = ENVS
other_args = OTHER_ARGS
@@ -10,7 +10,7 @@ from sglang.test.ascend.e2e.test_npu_performance_utils import (
from sglang.test.ci.ci_register import register_npu_ci
register_npu_ci(est_time=3600, suite="base-c-test-perf-2-npu-a3")
register_npu_ci(est_time=6500, suite="nightly-acc-2-npu-a3", nightly=True)
register_npu_ci(est_time=12000, suite="nightly-acc-2-npu-a3", nightly=True)
_is_pr_pipeline = os.environ.get("GITHUB_EVENT_NAME") == "pull_request"
@@ -20,7 +20,7 @@ ENVS = {
"HCCL_SOCKET_IFNAME": "lo",
"GLOO_SOCKET_IFNAME": "lo",
"HCCL_OP_EXPANSION_MODE": "AIV",
"HCCL_BUFFSIZE": "2000",
"DEEPEP_HCCL_BUFFSIZE": "2000",
}
OTHER_ARGS = [
@@ -52,7 +52,7 @@ OTHER_ARGS = [
]
class TestQwen3(TestNpuAccuracyTestCaseBase):
class TestQwen3_VL_8B_Thinking_MMMU(TestNpuAccuracyTestCaseBase):
model = QWEN3_VL_8B_THINKING_MODEL_PATH
envs = ENVS
other_args = OTHER_ARGS
@@ -53,6 +53,8 @@ KIMI_K2_6_OTHER_ARGS = [
6144,
"--max-prefill-tokens",
65536,
"--max-total-tokens",
32256,
"--enable-multimodal",
"--mm-attention-backend",
"ascend_attn",