Add infiniband to webportal job protocol schema (#4721)

microsoft · Jul 16, 2020 · 6abe345 · 6abe345
1 parent 4b9049f
commit 6abe345
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
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
@@ -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
@@ -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()