[AMD] Add more tests to per-commit-amd (#6926)

This commit is contained in:
Hubert Lu
2025-06-08 01:08:37 -07:00
committed by GitHub
parent 1fb76ebb93
commit 4740288303
3 changed files with 16 additions and 1 deletions

View File

@@ -91,10 +91,19 @@ suites = {
TestFile("test_w8a8_quantization.py", 46),
],
"per-commit-amd": [
TestFile("models/lora/test_lora_backend.py", 99),
TestFile("models/lora/test_multi_lora_backend.py", 60),
TestFile("models/lora/test_lora_cuda_graph.py", 250),
TestFile("test_mla.py", 242),
TestFile("test_mla_deepseek_v3.py", 221),
TestFile("test_torch_compile.py", 76),
TestFile("test_torch_compile_moe.py", 172),
TestFile("models/test_qwen_models.py", 82),
TestFile("models/test_reward_models.py", 132),
TestFile("test_abort.py", 51),
TestFile("test_block_int8.py", 22),
TestFile("test_create_kvindices.py", 2),
TestFile("test_chunked_prefill.py", 313),
],
"per-commit-2-gpu": [
TestFile("models/lora/test_lora_tp.py", 116),