[NVIDIA] Add CI workloads for GB200 (#12242)

This commit is contained in:
Kaixi Hou
2025-10-30 14:32:03 -07:00
committed by GitHub
parent 7d121448b2
commit c0d02cf4d1
5 changed files with 73 additions and 5 deletions
+3
View File
@@ -185,6 +185,9 @@ suites = {
TestFile("test_gpt_oss_4gpu.py", 600),
TestFile("test_llama31_fp4.py", 300),
],
"per-commit-4-gpu-gb200": [
TestFile("test_deepseek_v3_fp4_4gpu.py", 3600),
],
"per-commit-4-gpu-deepep": [
TestFile("ep/test_deepep_small.py", 531),
TestFile("ep/test_mooncake_ep_small.py", 450),