Skip to content

Commit 108248e

Browse files
authored
[TRTLLM-7999][infra] Add B300/GB300 single gpu test (#7951)
Signed-off-by: Yiqing Yan <[email protected]>
1 parent 7e2521a commit 108248e

File tree

3 files changed

+38
-0
lines changed

3 files changed

+38
-0
lines changed

jenkins/L0_Test.groovy

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2149,6 +2149,7 @@ def launchTestJobs(pipeline, testFilter)
21492149
fullSet = parallelJobs.keySet()
21502150

21512151
x86SlurmTestConfigs = [
2152+
"B300-PyTorch-1": ["b300-single", "l0_b300", 1, 1],
21522153
"DGX_B200-4_GPUs-PyTorch-1": ["b200-x4", "l0_dgx_b200", 1, 2, 4],
21532154
"DGX_B200-4_GPUs-PyTorch-2": ["b200-x4", "l0_dgx_b200", 2, 2, 4],
21542155
"DGX_B200-8_GPUs-PyTorch-1": ["b200-x8", "l0_dgx_b200", 1, 1, 8],
@@ -2181,6 +2182,7 @@ def launchTestJobs(pipeline, testFilter)
21812182
fullSet += SBSATestConfigs.keySet()
21822183

21832184
SBSASlurmTestConfigs = [
2185+
"GB300-PyTorch-1": ["gb300-single", "l0_gb300", 1, 1],
21842186
"GB200-4_GPUs-PyTorch-1": ["gb200-x4", "l0_gb200_multi_gpus", 1, 1, 4],
21852187
"GB200-4_GPUs-PyTorch-Post-Merge-1": ["gb200-x4", "l0_gb200_multi_gpus", 1, 1, 4],
21862188
"GB300-4_GPUs-PyTorch-Post-Merge-1": ["gb300-x4", "l0_gb300_multi_gpus", 1, 1, 4],
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
version: 0.0.1
2+
l0_b300:
3+
- condition:
4+
ranges:
5+
system_gpu_count:
6+
gte: 1
7+
lte: 1
8+
wildcards:
9+
gpu:
10+
- '*gb110*'
11+
linux_distribution_name: ubuntu*
12+
cpu: x86_64
13+
terms:
14+
stage: pre_merge
15+
backend: pytorch
16+
tests:
17+
# ------------- PyTorch tests ---------------
18+
- accuracy/test_llm_api_pytorch.py::TestLlama3_1_8B::test_nvfp4
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
version: 0.0.1
2+
l0_gb300:
3+
- condition:
4+
ranges:
5+
system_gpu_count:
6+
gte: 1
7+
lte: 1
8+
wildcards:
9+
gpu:
10+
- '*gb110*'
11+
linux_distribution_name: ubuntu*
12+
cpu: aarch64
13+
terms:
14+
stage: pre_merge
15+
backend: pytorch
16+
tests:
17+
# ------------- PyTorch tests ---------------
18+
- accuracy/test_llm_api_pytorch.py::TestQwen3_8B::test_w4a8_mxfp4[fp8-latency]

0 commit comments

Comments
 (0)