From dbad078d9584fb626fe7fc05dc8276e4f59e0050 Mon Sep 17 00:00:00 2001 From: Renan DelValle Date: Tue, 10 Sep 2019 15:00:13 -0700 Subject: [PATCH] Adding missing indirection for adding GPU requirements to task. --- job.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/job.go b/job.go index 800f98f..9dd99f6 100644 --- a/job.go +++ b/job.go @@ -136,6 +136,11 @@ func (j *AuroraJob) Disk(disk int64) *AuroraJob { return j } +func (j *AuroraJob) GPU(gpu int64) *AuroraJob { + j.task.GPU(gpu) + return j +} + func (j *AuroraJob) Tier(tier string) *AuroraJob { j.task.Tier(tier) return j