Skip to content

Commit f3c279a

Browse files
committed
Update vfio-pci validation to support the VFIO module used on Grace
Signed-off-by: Christopher Desiniotis <[email protected]>
1 parent c476d34 commit f3c279a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

cmd/nvidia-validator/main.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1541,7 +1541,10 @@ func (v *VfioPCI) runValidation() error {
15411541
}
15421542

15431543
for _, dev := range nvdevices {
1544-
if dev.Driver != "vfio-pci" {
1544+
// TODO: Do not hardcode a list of VFIO driver names. This would be possible if we
1545+
// added an API to go-nvlib which returns the most suitable VFIO driver for a GPU,
1546+
// using logic similar to https://github.com/NVIDIA/k8s-driver-manager/commit/874c8cd26d775db437f16a42c3e44e74301b3a35
1547+
if dev.Driver != "vfio-pci" && dev.Driver != "nvgrace_gpu_vfio_pci" {
15451548
return fmt.Errorf("device not bound to 'vfio-pci'; device: %s driver: '%s'", dev.Address, dev.Driver)
15461549
}
15471550
}

0 commit comments

Comments
 (0)