From c91a1b1dc8a172bb34d760892fbe13a53864aa39 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Fri, 24 Jan 2025 14:46:27 +0100 Subject: [PATCH] Add IMEX binaries to CDI discovery This change adds the nvidia-imex and nivdia-imex-ctl binaries to the list of driver binaries that are searched when using CDI. Signed-off-by: Evan Lezar --- pkg/nvcdi/driver-nvml.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pkg/nvcdi/driver-nvml.go b/pkg/nvcdi/driver-nvml.go index 007ead28..b0006aeb 100644 --- a/pkg/nvcdi/driver-nvml.go +++ b/pkg/nvcdi/driver-nvml.go @@ -184,6 +184,8 @@ func NewDriverBinariesDiscoverer(logger logger.Interface, driverRoot string) dis "nvidia-persistenced", /* Persistence mode utility */ "nvidia-cuda-mps-control", /* Multi process service CLI */ "nvidia-cuda-mps-server", /* Multi process service server */ + "nvidia-imex", /* NVIDIA IMEX Daemon */ + "nvidia-imex-ctl", /* NVIDIA IMEX control */ }, ) }