Bug fix for Thermos Payload generation relating to the GPU being set.

This commit is contained in:
Renan DelValle 2019-02-19 16:07:59 -08:00
parent 36fdc4abdb
commit e0ff0d137c
No known key found for this signature in database
GPG key ID: C240AD6D6F443EC9

View file

@ -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)
}