From 44c7bcc71e9646f1ffd97efb4c493c7f4b0b37e4 Mon Sep 17 00:00:00 2001
From: Yifan Xiong <yifan.xiong@microsoft.com>
Date: Tue, 8 Sep 2020 16:45:17 +0800
Subject: [PATCH] Use GPU number as SKU number when import config

Use GPU number as SKU number when import config.
---
 src/webportal/src/app/job-submission/models/job-task-role.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/webportal/src/app/job-submission/models/job-task-role.js b/src/webportal/src/app/job-submission/models/job-task-role.js
index b2305e3baa..0c12cccc89 100644
--- a/src/webportal/src/app/job-submission/models/job-task-role.js
+++ b/src/webportal/src/app/job-submission/models/job-task-role.js
@@ -110,7 +110,7 @@ export class JobTaskRole {
       hivedSku.skuNum = get(
         extras,
         `hivedScheduler.taskRoles.${name}.skuNum`,
-        1,
+        Math.max(get(resourcePerInstance, 'gpu', 1), 1),
       );
       hivedSku.skuType = get(
         extras,