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