Skip to content

Commit 94a53bc

Browse files
committed
Revert "fix(test): fixed node selector labels for E2Es with limiter"
This reverts commit b4f3388.
1 parent 4f4fac5 commit 94a53bc

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

test/e2e-saturation-based/e2e_limiter_test.go

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -38,23 +38,18 @@ func getGPUResourceName() corev1.ResourceName {
3838
}
3939

4040
// getGPUNodeSelector returns node selector for targeting specific GPU type.
41-
// Uses GPU product labels set in the Kind cluster with emulated GPU (see deploy/kind-emulator/setup.sh).
41+
// For nvidia (nvidia-mix cluster): targets H100 nodes
42+
// For amd (amd-mix cluster): targets MI300X nodes
4243
func getGPUNodeSelector() (map[string]string, string) {
4344
gpuType := os.Getenv("E2E_GPU_TYPE")
4445
if gpuType == "" {
4546
gpuType = "nvidia"
4647
}
4748

48-
// Use node labels set in the Kind cluster with emulated GPU (see deploy/kind-emulator/setup.sh).
49-
switch gpuType {
50-
case "nvidia":
51-
return map[string]string{"nvidia.com/gpu.product": "NVIDIA-A100-PCIE-80GB"}, "A100"
52-
case "amd":
53-
return map[string]string{"amd.com/gpu.product": "AMD-MI300X-192G"}, "MI300X"
54-
case "intel":
55-
return map[string]string{"intel.com/gpu.product": "Intel-Gaudi-2-96GB"}, "Gaudi2"
49+
if gpuType == "nvidia" {
50+
return map[string]string{"gpu-config": "4H100"}, "H100"
5651
}
57-
return map[string]string{}, ""
52+
return map[string]string{"gpu-config": "4MI300X"}, "MI300X"
5853
}
5954

6055
var _ = Describe("Test workload-variant-autoscaler - GPU Limiter Feature", Ordered, func() {

0 commit comments

Comments
 (0)