From a1c2f07b6eed0a96f387f59235638576382b6927 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Thu, 23 Jun 2022 16:01:04 +0200 Subject: [PATCH] Add /etc/cufile.json to list of required mounts Signed-off-by: Evan Lezar --- internal/discover/gds.go | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/internal/discover/gds.go b/internal/discover/gds.go index 6d8eef62..520b6e2e 100644 --- a/internal/discover/gds.go +++ b/internal/discover/gds.go @@ -36,22 +36,28 @@ func NewGDSDiscoverer(logger *logrus.Logger, root string) (Discover, error) { required: []string{"/dev/nvidia-fs*"}, } - mounts := &mounts{ + udev := &mounts{ logger: logger, lookup: lookup.NewDirectoryLocator(logger, root), required: []string{"/run/udev"}, } + cufile := &mounts{ + logger: logger, + lookup: lookup.NewFileLocator(logger, root), + required: []string{"/etc/cufile.json"}, + } + d := gdsDeviceDiscoverer{ logger: logger, devices: devices, - mounts: mounts, + mounts: NewList(udev, cufile), } return &d, nil } -// Devices discoveres the nvidia-fs device nodes for use with GPUDirect Storage +// Devices discovers the nvidia-fs device nodes for use with GPUDirect Storage func (d *gdsDeviceDiscoverer) Devices() ([]Device, error) { devicesAsMounts, err := d.devices.Mounts() if err != nil { @@ -66,7 +72,7 @@ func (d *gdsDeviceDiscoverer) Devices() ([]Device, error) { return devices, nil } -// Mounts discovers the required mounts for GDS. +// Mounts discovers the required mounts for GPUDirect Storage. // If no devices are discovered the discovered mounts are empty func (d *gdsDeviceDiscoverer) Mounts() ([]Mount, error) { devices, err := d.Devices()