diff --git a/jenkins/L0_Test.groovy b/jenkins/L0_Test.groovy index 28c5c182fb4..76f5c7c8708 100644 --- a/jenkins/L0_Test.groovy +++ b/jenkins/L0_Test.groovy @@ -664,6 +664,7 @@ def createKubernetesPodConfig(image, type, arch = "amd64", gpuCount = 1, perfMod selectors = """ kubernetes.io/arch: ${arch} kubernetes.io/os: linux + kubernetes.io/hostname: smc521ge-0081.ipp2a2.colossus nvidia.com/gpu_type: ${gpuType}""" } else if (perfMode && !hasMultipleGPUs) { // Not using the "perf" node currently due to hardware resource constraint. @@ -745,8 +746,6 @@ def createKubernetesPodConfig(image, type, arch = "amd64", gpuCount = 1, perfMod requiredDuringSchedulingIgnoredDuringExecution: nodeSelectorTerms: - matchExpressions: - - key: "tensorrt/taints" - operator: DoesNotExist - key: "tensorrt/affinity" operator: NotIn values: @@ -1843,9 +1842,9 @@ def launchTestJobs(pipeline, testFilter, dockerNode=null) "DGX_H200-4_GPUs-TensorRT-Post-Merge-1": ["dgx-h200-x4", "l0_dgx_h200", 1, 3, 4], "DGX_H200-4_GPUs-TensorRT-Post-Merge-2": ["dgx-h200-x4", "l0_dgx_h200", 2, 3, 4], "DGX_H200-4_GPUs-TensorRT-Post-Merge-3": ["dgx-h200-x4", "l0_dgx_h200", 3, 3, 4], - "RTXPro6000-Pytorch-Post-Merge-1": ["rtx-pro-6000", "l0_rtx_pro_6000", 1, 1], - "RTXPro6000-4_GPUs-Pytorch-Post-Merge-1": ["rtx-pro-6000-x4", "l0_rtx_pro_6000", 1, 2, 4], - "RTXPro6000-4_GPUs-Pytorch-Post-Merge-2": ["rtx-pro-6000-x4", "l0_rtx_pro_6000", 2, 2, 4], + //"RTXPro6000-Pytorch-Post-Merge-1": ["rtx-pro-6000", "l0_rtx_pro_6000", 1, 1], + "RTXPro6000-8_GPUs-Pytorch-Post-Merge-1": ["rtx-pro-6000-x8", "l0_rtx_pro_6000", 1, 1, 8], + //"RTXPro6000-4_GPUs-Pytorch-Post-Merge-2": ["rtx-pro-6000-x4", "l0_rtx_pro_6000", 2, 2, 4], ] parallelJobs = x86TestConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(LLM_DOCKER_IMAGE, values[0], "amd64", values[4] ?: 1, key.contains("Perf")), { diff --git a/tests/integration/test_lists/test-db/l0_rtx_pro_6000.yml b/tests/integration/test_lists/test-db/l0_rtx_pro_6000.yml index 49c29329dc7..b82bee9b4fd 100644 --- a/tests/integration/test_lists/test-db/l0_rtx_pro_6000.yml +++ b/tests/integration/test_lists/test-db/l0_rtx_pro_6000.yml @@ -34,8 +34,8 @@ l0_rtx_pro_6000: - condition: ranges: system_gpu_count: - gte: 4 - lte: 4 + gte: 8 + lte: 8 wildcards: gpu: - '*6000*'