UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-02-16 17:42:20 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add NewContainerEdits utility function

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2022-12-07 11:03:45 +01:00
parent 046d761f4c
commit 3140810c95
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/edits/edits.go
View File

@ -64,9 +64,7 @@ func FromDiscoverer(d discover.Discover) (*cdi.ContainerEdits, error) {
return nil, fmt.Errorf("failed to discover hooks: %v", err) return nil, fmt.Errorf("failed to discover hooks: %v", err)
} }
c := cdi.ContainerEdits{ c := NewContainerEdits()
ContainerEdits: &specs.ContainerEdits{},
}
for _, d := range devices { for _, d := range devices {
edits, err := device(d).toEdits() edits, err := device(d).toEdits()
if err != nil { if err != nil {
@ -83,7 +81,15 @@ func FromDiscoverer(d discover.Discover) (*cdi.ContainerEdits, error) {
c.Append(hook(h).toEdits()) c.Append(hook(h).toEdits())
} }
return &c, nil return c, nil
}
// NewContainerEdits is a utility function to create a CDI ContainerEdits struct.
func NewContainerEdits() *cdi.ContainerEdits {
c := cdi.ContainerEdits{
ContainerEdits: &specs.ContainerEdits{},
}
return &c
} }
// Modify applies the defined edits to the incoming OCI spec // Modify applies the defined edits to the incoming OCI spec