UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-06-26 18:18:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add update-nvidia-params hook to CDI spec
Some checks failed
CI Pipeline / code-scanning (push) Has been cancelled
Details
CI Pipeline / variables (push) Has been cancelled
Details
CI Pipeline / golang (push) Has been cancelled
Details
CI Pipeline / image (push) Has been cancelled
Details
CI Pipeline / e2e-test (push) Has been cancelled
Details

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2025-02-13 15:15:32 +01:00
parent 6396e19263
commit 21d23bc259
3 changed files with 30 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/nvcdi/api.go
View File

@@ -44,4 +44,7 @@ const (
// HookEnableCudaCompat refers to the hook used to enable CUDA Forward Compatibility.
// This was added with v1.17.5 of the NVIDIA Container Toolkit.
HookEnableCudaCompat = HookName("enable-cuda-compat")
// HookUpdateNvidiaParams refers to the hook used to ensure that device nodes
// are not created by nvidia-smi in a container.
HookUpdateNvidiaParams = HookName("update-nvidia-params")
)

8
pkg/nvcdi/driver-nvml.go
View File

@@ -115,6 +115,14 @@ func (l *nvcdilib) NewDriverLibraryDiscoverer(version string) (discover.Discover
updateLDCache, _ := discover.NewLDCacheUpdateHook(l.logger, libraries, l.nvidiaCDIHookPath, l.ldconfigPath)
discoverers = append(discoverers, updateLDCache)
if l.HookIsSupported(HookUpdateNvidiaParams) {
updateNvidiaParams := discover.CreateNvidiaCDIHook(
l.nvidiaCDIHookPath,
"update-nvidia-params",
)
discoverers = append(discoverers, updateNvidiaParams)
}
d := discover.Merge(discoverers...)
return d, nil