diff --git a/internal/discover/gds.go b/internal/discover/gds.go index cf762cd8a..d1f4b00bc 100644 --- a/internal/discover/gds.go +++ b/internal/discover/gds.go @@ -29,10 +29,11 @@ type gdsDeviceDiscoverer struct { } // NewGDSDiscoverer creates a discoverer for GPUDirect Storage devices and mounts. -func NewGDSDiscoverer(logger logger.Interface, driverRoot string, devRoot string) (Discover, error) { +func NewGDSDiscoverer(logger logger.Interface, driverRoot string) (Discover, error) { devices := NewCharDeviceDiscoverer( logger, - devRoot, + // The /dev/nvidia-fs* devices are always created at / + "/", []string{"/dev/nvidia-fs*"}, ) diff --git a/internal/modifier/gated.go b/internal/modifier/gated.go index a0239df8e..b88916dab 100644 --- a/internal/modifier/gated.go +++ b/internal/modifier/gated.go @@ -48,7 +48,7 @@ func NewFeatureGatedModifier(logger logger.Interface, cfg *config.Config, image devRoot := cfg.NVIDIAContainerCLIConfig.Root if image.Getenv("NVIDIA_GDS") == "enabled" { - d, err := discover.NewGDSDiscoverer(logger, driverRoot, devRoot) + d, err := discover.NewGDSDiscoverer(logger, driverRoot) if err != nil { return nil, fmt.Errorf("failed to construct discoverer for GDS devices: %w", err) } diff --git a/pkg/nvcdi/gds.go b/pkg/nvcdi/gds.go index 73892f967..16607f112 100644 --- a/pkg/nvcdi/gds.go +++ b/pkg/nvcdi/gds.go @@ -34,7 +34,7 @@ var _ Interface = (*gdslib)(nil) // GetAllDeviceSpecs returns the device specs for all available devices. func (l *gdslib) GetAllDeviceSpecs() ([]specs.Device, error) { - discoverer, err := discover.NewGDSDiscoverer(l.logger, l.driverRoot, l.devRoot) + discoverer, err := discover.NewGDSDiscoverer(l.logger, l.driverRoot) if err != nil { return nil, fmt.Errorf("failed to create GPUDirect Storage discoverer: %v", err) }