[AMD] Add test_fused_moe.py and test_rope_rocm.py to AMD CI (#5246)

This commit is contained in:
Hubert Lu
2025-07-06 01:47:16 -07:00
committed by GitHub
parent ea4bf12286
commit e00715eb66
3 changed files with 150 additions and 5 deletions

View File

@@ -125,6 +125,7 @@ suites = {
TestFile("test_chunked_prefill.py", 313),
TestFile("test_eval_fp8_accuracy.py", 303),
TestFile("test_function_call_parser.py", 10),
TestFile("test_fused_moe.py", 30),
TestFile("test_input_embeddings.py", 38),
TestFile("test_metrics.py", 32),
TestFile("test_no_chunked_prefill.py", 108),
@@ -142,6 +143,7 @@ suites = {
TestFile("test_vertex_endpoint.py", 31),
# TestFile("test_vision_chunked_prefill.py", 175), # Disabled temporarily and track in #7701
TestFile("test_reasoning_parser.py", 5),
TestFile("test_rope_rocm.py", 3),
],
"per-commit-npu": [
TestFile("test_ascend_attention_backend.py", 400),