diff --git a/form.yml.erb b/form.yml.erb index 9f60f240d5ffcf0eb51fac53d27c478dbb4931cc..fec9ffb86e823ef972658df9b5caaaa4184c645f 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 646da4103aa067e46e030eab27a4fbffc84da324..caeb57b0467ed984f3c80da67d0afe3eb88443f2 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"