From e0ff0d137c5b04bb408d75017f96c4aeb763ef36 Mon Sep 17 00:00:00 2001 From: Renan DelValle Date: Tue, 19 Feb 2019 16:07:59 -0800 Subject: [PATCH] Bug fix for Thermos Payload generation relating to the GPU being set. --- task.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/task.go b/task.go index c067371..5308f42 100644 --- a/task.go +++ b/task.go @@ -409,7 +409,7 @@ func (t *AuroraTask) BuildThermosPayload() error { t.thermos.disk(*t.resources[DISK].DiskMb) } - if t.resources[GPU].NumGpus != nil { + if t.resources[GPU]!= nil && t.resources[GPU].NumGpus != nil { t.thermos.gpu(*t.resources[GPU].NumGpus) }