Add CI for gpt-oss model on hopper (#8851)

This commit is contained in:
fzyzcjy
2025-08-09 15:34:23 +08:00
committed by GitHub
parent de8b8b6e5c
commit 442534aa44
7 changed files with 187 additions and 2 deletions

View File

@@ -63,6 +63,7 @@ suites = {
TestFile("test_fp8_kernel.py", 8),
TestFile("test_function_call_parser.py", 10),
TestFile("test_fused_moe.py", 30),
TestFile("test_gpt_oss_1gpu.py", 600),
TestFile("test_hicache.py", 116),
TestFile("test_hicache_mla.py", 127),
TestFile("test_hicache_storage.py", 127),
@@ -104,7 +105,7 @@ suites = {
TestFile("test_utils_update_weights.py", 48),
TestFile("test_vision_chunked_prefill.py", 175),
TestFile("test_vlm_input_format.py", 300),
TestFile("test_vision_openai_server_a.py", 584),
TestFile("test_vision_openai_server_a.py", 989),
TestFile("test_vision_openai_server_b.py", 620),
TestFile("test_w8a8_quantization.py", 46),
TestFile("test_reasoning_parser.py", 5),
@@ -176,6 +177,7 @@ suites = {
TestFile("test_update_weights_from_distributed.py", 103),
],
"per-commit-4-gpu": [
TestFile("test_gpt_oss_4gpu.py", 600),
TestFile("test_local_attn.py", 250),
TestFile("test_pp_single_node.py", 372),
TestFile("test_multi_instance_release_memory_occupation.py", 64),