diff --git a/jenkins/L0_Test.groovy b/jenkins/L0_Test.groovy index 342bed4e28..9a1011e80a 100644 --- a/jenkins/L0_Test.groovy +++ b/jenkins/L0_Test.groovy @@ -1930,6 +1930,7 @@ def launchTestJobs(pipeline, testFilter, dockerNode=null) x86SlurmTestConfigs = [ "DGX_B200-4_GPUs-PyTorch-Post-Merge-1": ["b200-x4", "l0_dgx_b200", 1, 1, 4], + "DGX_B300-4_GPUs-PyTorch-Post-Merge-1": ["b300-x4", "l0_dgx_b300", 1, 1, 4], ] fullSet += x86SlurmTestConfigs.keySet() @@ -1962,6 +1963,7 @@ def launchTestJobs(pipeline, testFilter, dockerNode=null) // "GB200-PyTorch-1": ["gb200-single", "l0_gb200", 1, 3], "GB200-4_GPUs-PyTorch-1": ["gb200-x4", "l0_gb200_multi_gpus", 1, 1, 4], "GB200-4_GPUs-PyTorch-Post-Merge-1": ["gb200-x4", "l0_gb200_multi_gpus", 1, 1, 4], + "GB300-4_GPUs-PyTorch-Post-Merge-1": ["gb300-x4", "l0_gb300_multi_gpus", 1, 1, 4], ] fullSet += SBSASlurmTestConfigs.keySet() diff --git a/tests/integration/test_lists/test-db/l0_dgx_b300.yml b/tests/integration/test_lists/test-db/l0_dgx_b300.yml new file mode 100644 index 0000000000..a6ecd6bf74 --- /dev/null +++ b/tests/integration/test_lists/test-db/l0_dgx_b300.yml @@ -0,0 +1,17 @@ +version: 0.0.1 +l0_dgx_b300: +- condition: + ranges: + system_gpu_count: + gte: 4 + lte: 4 + wildcards: + gpu: + - '*gb110*' + linux_distribution_name: ubuntu* + cpu: x86_64 + terms: + stage: post_merge + backend: pytorch + tests: + - unittest/_torch/multi_gpu_modeling -k "deepseek" diff --git a/tests/integration/test_lists/test-db/l0_gb300_multi_gpus.yml b/tests/integration/test_lists/test-db/l0_gb300_multi_gpus.yml new file mode 100644 index 0000000000..a4b49268f3 --- /dev/null +++ b/tests/integration/test_lists/test-db/l0_gb300_multi_gpus.yml @@ -0,0 +1,17 @@ +version: 0.0.1 +l0_gb300_multi_gpus: +- condition: + ranges: + system_gpu_count: + gte: 4 + lte: 4 + wildcards: + gpu: + - '*gb110*' + linux_distribution_name: ubuntu* + cpu: aarch64 + terms: + stage: post_merge + backend: pytorch + tests: + - unittest/_torch/multi_gpu_modeling -k "deepseek"