Merge pull request #8 from uabrc/Feat_add_amperenodes_partition

Adding amperenodes partition to JupyterLab interactive app

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

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"