From 6abe345cc757469c39ab5caf4bc7c5690be30d77 Mon Sep 17 00:00:00 2001
From: yiyione <yiyi@microsoft.com>
Date: Thu, 16 Jul 2020 15:03:52 +0800
Subject: [PATCH] Add infiniband to webportal job protocol schema (#4721)

---
 src/webportal/src/app/job-submission/models/job-task-role.js   | 2 ++
 src/webportal/src/app/job-submission/models/protocol-schema.js | 1 +
 2 files changed, 3 insertions(+)

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 677590828b..55b89a7a5b 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
@@ -98,6 +98,7 @@ export class JobTaskRole {
       output: get(taskRoleProtocol, 'output'),
       script: get(taskRoleProtocol, 'script'),
       shmMB: get(taskRoleProtocol, 'extraContainerOptions.shmMB'),
+      infiniband: get(taskRoleProtocol, 'extraContainerOptions.infiniband'),
     });
 
     const jobTaskRole = new JobTaskRole({
@@ -152,6 +153,7 @@ export class JobTaskRole {
       script: this.extraOptions.script,
       extraContainerOptions: removeEmptyProperties({
         shmMB: this.extraOptions.shmMB,
+        infiniband: this.extraOptions.infiniband,
       }),
       resourcePerInstance: resourcePerInstance,
       commands: isEmpty(this.commands)
diff --git a/src/webportal/src/app/job-submission/models/protocol-schema.js b/src/webportal/src/app/job-submission/models/protocol-schema.js
index f7d39c56b8..c406908828 100644
--- a/src/webportal/src/app/job-submission/models/protocol-schema.js
+++ b/src/webportal/src/app/job-submission/models/protocol-schema.js
@@ -48,6 +48,7 @@ export const taskRoleSchema = Joi.object().keys({
   script: Joi.string(),
   extraContainerOptions: Joi.object().keys({
     shmMB: Joi.number(),
+    infiniband: Joi.boolean(),
   }),
   resourcePerInstance: Joi.object()
     .required()