[CI] Ascend NPU CI enhancement (#8294)
Co-authored-by: ronnie_zheng <zl19940307@163.com>
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user