Remove redundant CDI annotations

When generating CDI specs for coherent and noncoherent devices,
we include CDI device annotations. This increases the minimum CDI spec
version to v0.6.0 which is not as widely supported as v0.5.0 or v0.3.0.

This change removes these annotations after splitting the generated
specs to produce coherent and/or non-coherent specs.

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

cmd/nvidia-ctk/cdi/generate/generate.go

@@ -447,6 +447,7 @@ type deviceSpecs []specs.Device
 func (d deviceSpecs) splitOnAnnotation(key string) map[string][]specs.Device {
 	splitSpecs := make(map[string][]specs.Device)
 
+	var specsToRemoveAnnotations []*specs.Device
 	for _, deviceSpec := range d {
 		if len(deviceSpec.Annotations) == 0 {
 			continue
@@ -456,6 +457,21 @@ func (d deviceSpecs) splitOnAnnotation(key string) map[string][]specs.Device {
 			continue
 		}
 		splitSpecs[key+"="+value] = append(splitSpecs[key+"="+value], deviceSpec)
+		specsToRemoveAnnotations = append(specsToRemoveAnnotations, &deviceSpec)
+	}
+
+	// We also remove the annotations that were used to split the devices:
+	for _, deviceSpec := range specsToRemoveAnnotations {
+		if len(deviceSpec.Annotations) == 0 {
+			continue
+		}
+		if _, ok := deviceSpec.Annotations[key]; !ok {
+			continue
+		}
+		delete(deviceSpec.Annotations, key)
+		if len(deviceSpec.Annotations) == 0 {
+			deviceSpec.Annotations = nil
+		}
 	}
 
 	return splitSpecs