Merge pull request #2 from ananaysingh/killTaskAPI
GPU allocation changes
This commit is contained in:
commit
ff6c3e1df6
1 changed files with 4 additions and 1 deletions
|
@ -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)
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue