Skip to content

Commit 1bc719a

Browse files
committed
Add b300 tests
Signed-off-by: Yiqing Yan <[email protected]>
1 parent ba0a86e commit 1bc719a

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

jenkins/L0_Test.groovy

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -656,6 +656,9 @@ def createKubernetesPodConfig(image, type, arch = "amd64", gpuCount = 1, perfMod
656656
}
657657

658658
def gpuType = KubernetesManager.selectGPU(type)
659+
if (type.contains("b300")) {
660+
gpuType = "NVIDIA_HGX_B300"
661+
}
659662
nodeLabelPrefix = type
660663

661664
targetCould = "kubernetes"
@@ -1843,6 +1846,7 @@ def launchTestJobs(pipeline, testFilter, dockerNode=null)
18431846
"DGX_H200-4_GPUs-TensorRT-Post-Merge-1": ["dgx-h200-x4", "l0_dgx_h200", 1, 3, 4],
18441847
"DGX_H200-4_GPUs-TensorRT-Post-Merge-2": ["dgx-h200-x4", "l0_dgx_h200", 2, 3, 4],
18451848
"DGX_H200-4_GPUs-TensorRT-Post-Merge-3": ["dgx-h200-x4", "l0_dgx_h200", 3, 3, 4],
1849+
"B300_PCIe-PyTorch-1": ["b300", "l0_b300", 1, 1],
18461850
]
18471851

18481852
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_b200:
3+
- condition:
4+
ranges:
5+
system_gpu_count:
6+
gte: 1
7+
lte: 1
8+
wildcards:
9+
gpu:
10+
- '*b300*'
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)