Allow opt-out of all CDI device

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

Author: elezar

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

@@ -75,6 +75,8 @@ type options struct {
 
 	deviceIDs []string
 
+	noAllDevice bool
+
 	// the following are used for dependency injection during spec generation.
 	nvmllib nvml.Interface
 }
@@ -242,6 +244,12 @@ func (m command) build() *cli.Command {
 				Destination: &opts.deviceIDs,
 				Sources:     cli.EnvVars("NVIDIA_CTK_CDI_GENERATE_DEVICE_IDS"),
 			},
+			&cli.BoolFlag{
+				Name:        "no-all-device",
+				Usage:       "Don't generate an `all` device for the resultant spec",
+				Destination: &opts.noAllDevice,
+				Sources:     cli.EnvVars("NVIDIA_CTK_CDI_GENERATE_NO_ALL_DEVICE"),
+			},
 		},
 	}
 
@@ -397,13 +405,18 @@ func (m command) generateSpecs(opts *options) ([]generatedSpecs, error) {
 		spec.WithVendor(opts.vendor),
 		spec.WithEdits(*commonEdits.ContainerEdits),
 		spec.WithFormat(opts.format),
-		spec.WithMergedDeviceOptions(
-			transform.WithName(allDeviceName),
-			transform.WithSkipIfExists(true),
-		),
 		spec.WithPermissions(0644),
 	}
 
+	if !opts.noAllDevice {
+		commonSpecOptions = append(commonSpecOptions,
+			spec.WithMergedDeviceOptions(
+				transform.WithName(allDeviceName),
+				transform.WithSkipIfExists(true),
+			),
+		)
+	}
+
 	fullSpec, err := spec.New(
 		append(commonSpecOptions,
 			spec.WithClass(opts.class),