[CI] Ascend NPU CI enhancement (#8294)

Co-authored-by: ronnie_zheng <zl19940307@163.com>
This commit is contained in:
Even Zhou
2025-08-04 13:16:38 +08:00
committed by GitHub
parent f57d2dc162
commit fee0ab0fba
9 changed files with 415 additions and 189 deletions

View File

@@ -154,8 +154,14 @@ suites = {
TestFile("test_rope_rocm.py", 3),
TestFile("test_awq_dequant.py", 2),
],
"per-commit-npu": [
TestFile("test_ascend_attention_backend.py", 400),
"per-commit-1-ascend-npu": [
TestFile("test_ascend_tp1_bf16.py", 400),
],
"per-commit-2-ascend-npu": [
TestFile("test_ascend_tp2_bf16.py", 400),
],
"per-commit-4-ascend-npu": [
TestFile("test_ascend_mla_w8a8int8.py", 400),
],
"per-commit-2-gpu": [
TestFile("models/lora/test_lora_tp.py", 116),