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 cdi mode to NVIDIA Container Runtime

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2022-06-20 17:03:02 +02:00
parent beff276a52
commit 404e266222
2 changed files with 102 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/nvidia-container-runtime/runtime_factory.go
View File

@@ -91,6 +91,8 @@ func newModeModifier(logger *logrus.Logger, cfg *config.Config, ociSpec oci.Spec
return modifier.NewStableRuntimeModifier(logger), nil
case "csv":
return modifier.NewCSVModifier(logger, cfg, ociSpec)
case "cdi":
return modifier.NewCDIModifier(logger, cfg, ociSpec)
}
return nil, fmt.Errorf("invalid runtime mode: %v", cfg.NVIDIAContainerRuntimeConfig.Mode)