From b1847f18c10964b33d984ba8804a95d18561f1dc Mon Sep 17 00:00:00 2001
From: ananaysingh <54039499+ananaysingh@users.noreply.github.com>
Date: Tue, 16 Aug 2022 22:19:50 +0530
Subject: [PATCH] GPU allocation changes

---
 internal/job.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/internal/job.go b/internal/job.go
index 2f5e09b..b2ffc8b 100644
--- a/internal/job.go
+++ b/internal/job.go
@@ -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)
 	}