Allow nvcdi FeatureFlags to be configured for jit-cdi mode

This change allows users the posibility to explicitly specify
feature flags for using the `jit-cdi` mode. This allows, for
example, for users to opt-in to use nvsandboxutils in the
default mode in addition to when generating CDI specs
explicitly.

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

Author: elezar

internal/config/runtime.go

@@ -16,6 +16,8 @@
 
 package config
 
+import "github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi"
+
 // RuntimeConfig stores the config options for the NVIDIA Container Runtime
 type RuntimeConfig struct {
 	DebugFilePath string `toml:"debug"`
@@ -31,6 +33,7 @@ type RuntimeConfig struct {
 type modesConfig struct {
 	CSV    csvModeConfig    `toml:"csv"`
 	CDI    cdiModeConfig    `toml:"cdi"`
+	JitCDI jitCDIModeConfig `toml:"jit-cdi,omitempty"`
 	Legacy legacyModeConfig `toml:"legacy"`
 }
 
@@ -43,6 +46,11 @@ type cdiModeConfig struct {
 	AnnotationPrefixes []string `toml:"annotation-prefixes"`
 }
 
+type jitCDIModeConfig struct {
+	// NVCDIFeatureFlags sets a list of nvcdi features explicitly.
+	NVCDIFeatureFlags []nvcdi.FeatureFlag `toml:"nvcdi-feature-flags,omitempty"`
+}
+
 type csvModeConfig struct {
 	MountSpecPath string `toml:"mount-spec-path"`
 }

internal/modifier/cdi.go

@@ -192,6 +192,7 @@ func newAutomaticCDISpecModifier(logger logger.Interface, cfg *config.Config, de
 			nvcdi.WithVendor(automaticDeviceVendor),
 			nvcdi.WithClass(perModeDeviceClass[mode]),
 			nvcdi.WithMode(mode),
+			nvcdi.WithFeatureFlags(cfg.NVIDIAContainerRuntimeConfig.Modes.JitCDI.NVCDIFeatureFlags...),
 		)
 		if err != nil {
 			return nil, fmt.Errorf("failed to construct CDI library for mode %q: %w", mode, err)