File tree Expand file tree Collapse file tree
test/e2e-saturation-based Expand file tree Collapse file tree Original file line number Diff line number Diff 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
4243func 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
6055var _ = Describe ("Test workload-variant-autoscaler - GPU Limiter Feature" , Ordered , func () {
You can’t perform that action at this time.
0 commit comments