Add mode option to nvcdi API

Signed-off-by: Evan Lezar <elezar@nvidia.com>

pkg/nvcdi/lib.go

@@ -28,6 +28,7 @@ import (
 type nvcdilib struct {
 	logger        *logrus.Logger
 	nvmllib       nvml.Interface
+	mode          string
 	devicelib     device.Interface
 	deviceNamer   DeviceNamer
 	driverRoot    string

pkg/nvcdi/options.go

@@ -66,3 +66,10 @@ func WithNvmlLib(nvmllib nvml.Interface) Option {
 		l.nvmllib = nvmllib
 	}
 }
+
+// WithMode sets the discovery mode for the library
+func WithMode(mode string) Option {
+	return func(l *nvcdilib) {
+		l.mode = mode
+	}
+}