Merge pull request #102 from jwcesign/fix-flannel

fix: fix gpu node label
cloudpilot-ai · Nov 7, 2024 · e82e20f · e82e20f
2 parents 142c8b6 + daa8fad
commit e82e20f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/providers/instancetype/types.go b/pkg/providers/instancetype/types.go
@@ -154,7 +154,7 @@ func computeRequirements(info *ecsclient.DescribeInstanceTypesResponseBodyInstan
 		requirements.Get(v1alpha1.LabelInstanceGPUName).Insert(lowerKabobCase(*info.GPUSpec))
 		requirements.Get(v1alpha1.LabelInstanceGPUManufacturer).Insert(getGPUManufacturer(*info.GPUSpec))
 		requirements.Get(v1alpha1.LabelInstanceGPUCount).Insert(fmt.Sprint(*info.GPUAmount))
-		requirements.Get(v1alpha1.LabelInstanceGPUMemory).Insert(fmt.Sprint(info.GPUMemorySize))
+		requirements.Get(v1alpha1.LabelInstanceGPUMemory).Insert(fmt.Sprint(*info.GPUMemorySize))
 	}
 
 	// CPU Manufacturer, valid options: intel, amd
@@ -316,7 +316,9 @@ func amdGPUs(info *ecsclient.DescribeInstanceTypesResponseBodyInstanceTypesInsta
 }
 
 func lowerKabobCase(s string) string {
-	return strings.ToLower(strings.ReplaceAll(s, " ", "-"))
+	ret := strings.ReplaceAll(s, " ", "-")
+	ret = strings.ReplaceAll(ret, "/", "-")
+	return strings.ToLower(strings.ReplaceAll(ret, "*", "-"))
 }
 
 func getGPUManufacturer(gpuName string) string {