Skip to content

Commit

Permalink
restructure deviceRequest assignment
Browse files Browse the repository at this point in the history
  • Loading branch information
prateekchaudhry committed Sep 27, 2021
1 parent 0e687b5 commit ea75eec
Showing 1 changed file with 6 additions and 13 deletions.
19 changes: 6 additions & 13 deletions agent/api/task/task.go
Expand Up @@ -1560,27 +1560,20 @@ func (task *Task) getDockerResources(container *apicontainer.Container, cfg *con
task.Arn, container.Name, apicontainer.DockerContainerMinimumMemoryInBytes)
dockerMem = apicontainer.DockerContainerMinimumMemoryInBytes
}
resources := dockercontainer.Resources{}
// Set CPUShares
cpuShare := task.dockerCPUShares(container.CPU)
resources := dockercontainer.Resources{
Memory: dockerMem,
CPUShares: cpuShare,
}
if cfg.External.Enabled() && cfg.GPUSupportEnabled {
deviceRequest := dockercontainer.DeviceRequest{
Capabilities: [][]string{[]string{"gpu"}},
DeviceIDs: container.GPUIDs,
}
resources = dockercontainer.Resources{
Memory: dockerMem,
CPUShares: cpuShare,
DeviceRequests: []dockercontainer.DeviceRequest{deviceRequest},
}
return resources
} else {
resources = dockercontainer.Resources{
Memory: dockerMem,
CPUShares: cpuShare,
}
return resources
resources.DeviceRequests = []dockercontainer.DeviceRequest{deviceRequest}
}
return resources
}

// shouldOverrideNetworkMode returns true if the network mode of the container needs
Expand Down

0 comments on commit ea75eec

Please sign in to comment.