From 4cc07e8212ff5d5853040dc5b1e65e32097449db Mon Sep 17 00:00:00 2001
From: Ravi Tripathi <ravi89@uab.edu>
Date: Tue, 13 Jun 2023 14:36:40 -0500
Subject: [PATCH] Adding amperenodes partition to JupyterLab interactive app
---
form.yml.erb | 11 +++++++++++
submit.yml.erb | 4 ++--
2 files changed, 13 insertions(+), 2 deletions(-)
diff --git a/form.yml.erb b/form.yml.erb
index 9f60f24..fec9ffb 100644
--- a/form.yml.erb
+++ b/form.yml.erb
@@ -80,6 +80,13 @@ attributes:
max: 128
step: 1
+ bc_num_gpus:
+ label: Number of GPUs
+ value: 1
+ min: 1
+ max: 4
+ step: 1
+
bc_partition:
widget: select
label: Partition
@@ -95,6 +102,9 @@ attributes:
- [ "largemem-long", "largemem-long" ]
- [ "amd-hdr100", "amd-hdr100" ]
- [ "intel-dcb", "intel-dcb" ]
+ - [ "amperenodes-debug", "amperenodes-debug" ]
+ - [ "amperenodes", "amperenodes" ]
+ - [ "amperenodes-medium", "amperenodes-medium" ]
# All of the attributes that make up the Dashboard form (in respective order),
# and made available to the submit configuration file and the template ERB
@@ -111,6 +121,7 @@ form:
- extra_jupyter_args
- bc_num_hours
- bc_partition
+ - bc_num_gpus
- bc_num_slots
- bc_num_mems
- bc_email_on_started
diff --git a/submit.yml.erb b/submit.yml.erb
index 646da41..caeb57b 100644
--- a/submit.yml.erb
+++ b/submit.yml.erb
@@ -25,8 +25,8 @@ script:
- "--partition=<%= bc_partition %>"
- "--time=<%= bc_num_hours.blank? ? 1 : bc_num_hours.to_i %>:00:00"
- "--job-name=ood-jupyterlab"
-<%- if bc_partition.include? "pascalnodes" -%>
- - "--gres=gpu:1"
+<%- if bc_partition.include? "pascalnodes" or bc_partition.include? "amperenodes" -%>
+ - "--gres=gpu:<%= bc_num_gpus.blank? ? 1 : bc_num_gpus.to_i %>"
<%- end -%>
<%- if bc_email_on_started == "1" -%>
- "--mail-type=BEGIN"
--
GitLab