Skip to content

Commit 8a0bedf

Browse files
committed
Add b300 tests
Signed-off-by: Yiqing Yan <[email protected]>
1 parent 6e13160 commit 8a0bedf

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

jenkins/L0_Test.groovy

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
@Library(['bloom-jenkins-shared-lib@main', 'trtllm-jenkins-shared-lib@main']) _
1+
@Library(['bloom-jenkins-shared-lib@user/yiqingy/dgx_b300', 'trtllm-jenkins-shared-lib@main']) _
22

33
import java.lang.InterruptedException
44
import groovy.transform.Field
@@ -1876,6 +1876,7 @@ def launchTestJobs(pipeline, testFilter, dockerNode=null)
18761876
"RTXPro6000-Pytorch-Post-Merge-1": ["rtx-pro-6000", "l0_rtx_pro_6000", 1, 1],
18771877
"RTXPro6000-4_GPUs-Pytorch-Post-Merge-1": ["rtx-pro-6000-x4", "l0_rtx_pro_6000", 1, 2, 4],
18781878
"RTXPro6000-4_GPUs-Pytorch-Post-Merge-2": ["rtx-pro-6000-x4", "l0_rtx_pro_6000", 2, 2, 4],
1879+
"DGX_B300-4_GPUs-PyTorch-1": ["dgx-b300-x4", "l0_dgx_b300", 1, 1, 4],
18791880
]
18801881

18811882
parallelJobs = x86TestConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(LLM_DOCKER_IMAGE, values[0], "amd64", values[4] ?: 1, key.contains("Perf")), {
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
version: 0.0.1
2+
l0_dgx_b300:
3+
- condition:
4+
ranges:
5+
system_gpu_count:
6+
gte: 4
7+
lte: 4
8+
wildcards:
9+
gpu:
10+
- '*gb110*'
11+
linux_distribution_name: ubuntu*
12+
terms:
13+
stage: pre_merge
14+
backend: pytorch
15+
tests:
16+
# ------------- PyTorch tests ---------------
17+
- unittest/_torch/multi_gpu_modeling -k "deepseek"

0 commit comments

Comments
 (0)