From e330a938fd68184aa3c3a943ae25520264bb8f54 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Thu, 6 Feb 2025 23:17:21 +0100 Subject: [PATCH] Add enable-cuda-compat hook to CDI spec generation Signed-off-by: Evan Lezar --- pkg/nvcdi/driver-nvml.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pkg/nvcdi/driver-nvml.go b/pkg/nvcdi/driver-nvml.go index 007ead28..b61a6c65 100644 --- a/pkg/nvcdi/driver-nvml.go +++ b/pkg/nvcdi/driver-nvml.go @@ -97,6 +97,8 @@ func NewDriverLibraryDiscoverer(logger logger.Interface, driver *root.Driver, nv libraryPaths, ) + // TODO: The following should use the version directly. + cudaCompatLibHookDiscoverer := discover.NewCUDACompatHookDiscoverer(logger, nvidiaCDIHookPath, driver) updateLDCache, _ := discover.NewLDCacheUpdateHook(logger, libraries, nvidiaCDIHookPath, ldconfigPath) d := discover.Merge( @@ -105,6 +107,7 @@ func NewDriverLibraryDiscoverer(logger logger.Interface, driver *root.Driver, nv version, nvidiaCDIHookPath, ), + cudaCompatLibHookDiscoverer, updateLDCache, )