/
defaults.go
48 lines (40 loc) · 1.27 KB
/
defaults.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
package model
import (
"github.com/determined-ai/determined/master/pkg/schemas/expconf"
)
// DefaultResourcesConfig returns the default resources configuration.
func DefaultResourcesConfig(taskContainerDefaults *TaskContainerDefaultsConfig) ResourcesConfig {
config := ResourcesConfig{
Weight: 1,
NativeParallel: false,
}
if taskContainerDefaults == nil {
return config
}
config.Devices = taskContainerDefaults.Devices
return config
}
// DefaultEnvConfig returns the default environment configuration.
func DefaultEnvConfig(taskContainerDefaults *TaskContainerDefaultsConfig) Environment {
config := Environment{
Image: RuntimeItem{
CPU: expconf.CPUImage,
CUDA: expconf.CUDAImage,
ROCM: expconf.ROCMImage,
},
}
if taskContainerDefaults == nil {
return config
}
config.RegistryAuth = taskContainerDefaults.RegistryAuth
config.ForcePullImage = taskContainerDefaults.ForcePullImage
if taskContainerDefaults.Image != nil {
config.Image = *taskContainerDefaults.Image
}
if taskContainerDefaults.EnvironmentVariables != nil {
config.EnvironmentVariables = *taskContainerDefaults.EnvironmentVariables
}
config.AddCapabilities = taskContainerDefaults.AddCapabilities
config.DropCapabilities = taskContainerDefaults.DropCapabilities
return config
}