[NPU] Adapt pr-gate for pr-test workflow & workflows refresh (#13567)

This commit is contained in:
Even Zhou
2025-11-19 21:55:17 +08:00
committed by GitHub
parent 97ba2c2d1e
commit d4a4dcdfb3
6 changed files with 67 additions and 45 deletions

View File

@@ -514,25 +514,26 @@ suite_xpu = {
}
# Add Ascend NPU tests
# TODO: Set accurate estimate time
# NOTE: please sort the test cases alphabetically by the test file name
suite_ascend = {
"per-commit-1-ascend-npu": [
"per-commit-1-npu-a2": [
TestFile("ascend/test_ascend_graph_tp1_bf16.py", 400),
TestFile("ascend/test_ascend_tp1_bf16.py", 400),
TestFile("ascend/test_ascend_hicache_mha.py", 400),
TestFile("ascend/test_ascend_sampling_backend.py", 400),
TestFile("ascend/test_ascend_tp1_bf16.py", 400),
],
"per-commit-2-ascend-npu": [
"per-commit-2-npu-a2": [
TestFile("ascend/test_ascend_graph_tp2_bf16.py", 400),
TestFile("ascend/test_ascend_mla_fia_w8a8int8.py", 400),
TestFile("ascend/test_ascend_tp2_bf16.py", 400),
TestFile("ascend/test_ascend_tp2_fia_bf16.py", 400),
],
"per-commit-4-ascend-npu": [
"per-commit-4-npu-a2": [
TestFile("ascend/test_ascend_mla_w8a8int8.py", 400),
TestFile("ascend/test_ascend_tp4_bf16.py", 400),
],
"per-commit-16-ascend-a3": [
"per-commit-16-npu-a3": [
TestFile("ascend/test_ascend_deepep.py", 400),
TestFile("ascend/test_ascend_deepseek_mtp.py", 400),
],