GPU allocation changes

This commit is contained in:
ananaysingh 2022-08-16 22:19:50 +05:30
parent db2d4e9e17
commit b1847f18c1

View file

@ -93,7 +93,6 @@ func (j *Job) ToRealis() (*realis.AuroraJob, error) {
RAM(j.RAM). RAM(j.RAM).
Disk(j.Disk). Disk(j.Disk).
AddPorts(int(j.Port)). AddPorts(int(j.Port)).
GPU(j.GPU).
IsService(j.Service). IsService(j.Service).
Tier(j.Tier). Tier(j.Tier).
Priority(j.Priority). Priority(j.Priority).
@ -101,6 +100,10 @@ func (j *Job) ToRealis() (*realis.AuroraJob, error) {
InstanceCount(j.Instances). InstanceCount(j.Instances).
MaxFailure(j.MaxFailures) MaxFailure(j.MaxFailures)
if j.GPU > 0 {
auroraJob.GPU(j.GPU)
}
if j.CronSchedule != nil { if j.CronSchedule != nil {
auroraJob.CronSchedule(*j.CronSchedule) auroraJob.CronSchedule(*j.CronSchedule)
} }