mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2024-11-24 04:54:00 +00:00
b18ac09f77
This change consolidates the handling of NVIDIA_DRIVER_CAPABILITIES in the interal/image package. Signed-off-by: Evan Lezar <elezar@nvidia.com>
28 lines
422 B
Go
28 lines
422 B
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
)
|
|
|
|
func capabilityToCLI(cap string) string {
|
|
switch cap {
|
|
case "compute":
|
|
return "--compute"
|
|
case "compat32":
|
|
return "--compat32"
|
|
case "graphics":
|
|
return "--graphics"
|
|
case "utility":
|
|
return "--utility"
|
|
case "video":
|
|
return "--video"
|
|
case "display":
|
|
return "--display"
|
|
case "ngx":
|
|
return "--ngx"
|
|
default:
|
|
log.Panicln("unknown driver capability:", cap)
|
|
}
|
|
return ""
|
|
}
|