mirror of
https://github.com/docker/compose.git
synced 2026-10-22 20:02:25 +00:00
add support for CDI device request using devices
Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>
This commit is contained in:
parent
f794c79eb3
commit
3a95a0872d
2 changed files with 16 additions and 1 deletions
2
go.mod
2
go.mod
|
|
@ -57,6 +57,7 @@ require (
|
|||
google.golang.org/grpc v1.62.0
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
gotest.tools/v3 v3.5.1
|
||||
tags.cncf.io/container-device-interface v0.8.0
|
||||
)
|
||||
|
||||
require (
|
||||
|
|
@ -190,5 +191,4 @@ require (
|
|||
sigs.k8s.io/json v0.0.0-20221116044647-bc3834ca7abd // indirect
|
||||
sigs.k8s.io/structured-merge-diff/v4 v4.4.1 // indirect
|
||||
sigs.k8s.io/yaml v1.3.0 // indirect
|
||||
tags.cncf.io/container-device-interface v0.8.0 // indirect
|
||||
)
|
||||
|
|
|
|||
|
|
@ -48,6 +48,7 @@ import (
|
|||
"github.com/docker/docker/errdefs"
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/sirupsen/logrus"
|
||||
cdi "tags.cncf.io/container-device-interface/pkg/parser"
|
||||
)
|
||||
|
||||
type createOptions struct {
|
||||
|
|
@ -645,7 +646,14 @@ func getDeployResources(s types.ServiceConfig) container.Resources {
|
|||
setReservations(s.Deploy.Resources.Reservations, &resources)
|
||||
}
|
||||
|
||||
var cdiDeviceNames []string
|
||||
for _, device := range s.Devices {
|
||||
|
||||
if device.Source == device.Target && cdi.IsQualifiedName(device.Source) {
|
||||
cdiDeviceNames = append(cdiDeviceNames, device.Source)
|
||||
continue
|
||||
}
|
||||
|
||||
resources.Devices = append(resources.Devices, container.DeviceMapping{
|
||||
PathOnHost: device.Source,
|
||||
PathInContainer: device.Target,
|
||||
|
|
@ -653,6 +661,13 @@ func getDeployResources(s types.ServiceConfig) container.Resources {
|
|||
})
|
||||
}
|
||||
|
||||
if len(cdiDeviceNames) > 0 {
|
||||
resources.DeviceRequests = append(resources.DeviceRequests, container.DeviceRequest{
|
||||
Driver: "cdi",
|
||||
DeviceIDs: cdiDeviceNames,
|
||||
})
|
||||
}
|
||||
|
||||
ulimits := toUlimits(s.Ulimits)
|
||||
resources.Ulimits = ulimits
|
||||
return resources
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue