UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-22 08:18:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Handle case were runtime name is set to predefined name

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2023-03-07 20:59:22 +02:00
parent 62d88e7c95
commit cc7a6f166b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tools/container/operator/operator.go
View File

@ -84,8 +84,13 @@ func (r *Runtimes) add(runtime Runtime) {
// nvidiaRuntime creates a runtime that corresponds to the nvidia runtime. // nvidiaRuntime creates a runtime that corresponds to the nvidia runtime.
// If the nvidiaRuntimeName is specified, this name us used unless this is exactly equal to nvidia-experimental. // If the nvidiaRuntimeName is specified, this name us used unless this is exactly equal to nvidia-experimental.
func (c config) nvidiaRuntime() Runtime { func (c config) nvidiaRuntime() Runtime {
predefinedRuntimes := map[string]struct{}{
"nvidia-experimental": {},
"nvidia-cdi": {},
"nvidia-legacy": {},
}
name := c.nvidiaRuntimeName name := c.nvidiaRuntimeName
if name == experimentalRuntimeName { if _, isPredefinedRuntime := predefinedRuntimes[name]; isPredefinedRuntime {
name = defaultRuntimeName name = defaultRuntimeName
} }
return c.newRuntime(name, "nvidia-container-runtime") return c.newRuntime(name, "nvidia-container-runtime")