From 069cab589aea57b281c8dae3b4420973bdbd4967 Mon Sep 17 00:00:00 2001
From: Yifan Xiong <yifan.xiong@microsoft.com>
Date: Wed, 9 Sep 2020 10:31:19 +0800
Subject: [PATCH] Use GPU number as SKU number when import config (#4883)

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,