Merge pull request #1226 from racheljpg/gputypevalue

OCPCLOUD-2501: Update to set GpuType annotation to remove hardcoded value
openshift · Apr 26, 2024 · ac84cec · ac84cec
2 parents cf605b0 + b6f73b6
commit ac84cec
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pkg/util/machineset/util.go b/pkg/util/machineset/util.go
@@ -77,9 +77,8 @@ func SetGpuCountAnnotation(annotations map[string]string, value string) map[stri
 
 // SetGpuTypeAnnotation sets a value for gpu type in the annotations of a MachineSet.
 // Currently, we only support nvidia as a gpu type.
-func SetGpuTypeAnnotation(annotations map[string]string, _ string) map[string]string {
-	// TODO: Once we introduce proper gpu types, this needs to be changed.
-	annotations[GpuTypeKey] = GpuNvidiaType
+func SetGpuTypeAnnotation(annotations map[string]string, value string) map[string]string {
+	annotations[GpuTypeKey] = value
 
 	return annotations
 }