Add support for nvswitch mode to nvcdi API

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

Author: elezar

pkg/nvcdi/gated.go

@@ -61,6 +61,8 @@ func (l *gatedlib) getModeDiscoverer() (discover.Discover, error) {
 		return discover.NewGDSDiscoverer(l.logger, l.driverRoot, l.devRoot)
 	case ModeMofed:
 		return discover.NewMOFEDDiscoverer(l.logger, l.driverRoot)
+	case ModeNvswitch:
+		return discover.NewNvSwitchDiscoverer(l.logger, l.devRoot)
 	default:
 		return nil, fmt.Errorf("unrecognized mode")
 	}

pkg/nvcdi/mode.go

@@ -42,8 +42,10 @@ const (
 	// ModeCSV configures the CDI spec generator to generate a spec based on the contents of CSV
 	// mountspec files.
 	ModeCSV = Mode("csv")
-	// ModeImex configures the CDI spec generated to generate a spec for the available IMEX channels.
+	// ModeImex configures the CDI spec generator to generate a spec for the available IMEX channels.
 	ModeImex = Mode("imex")
+	// ModeNvswitch configures the CDI spec generator to generate a spec for the available nvswitch devices.
+	ModeNvswitch = Mode("nvswitch")
 )
 
 type modeConstraint interface {
@@ -69,6 +71,7 @@ func getModes() modes {
 			ModeManagement,
 			ModeMofed,
 			ModeNvml,
+			ModeNvswitch,
 			ModeWsl,
 		}
 		lookup := make(map[Mode]bool)