Add CI for gpt-oss model on hopper (#8851)
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user