diff --git a/form.yml.erb b/form.yml.erb index f2ef4c0f3d5b018736128c08060f430bc5aa01cc..f3c441a63db3bb903bc332b5b0684cb49c08868a 100644 --- a/form.yml.erb +++ b/form.yml.erb @@ -30,6 +30,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 @@ -45,6 +52,9 @@ attributes: - [ "largemem-long", "largemem-long" ] - [ "amd-hdr100", "amd-hdr100" ] - [ "intel-dcb", "intel-dcb" ] + - [ "amperenodes-debug", "amperenodes-debug" ] + - [ "amperenodes", "amperenodes" ] + - [ "amperenodes-medium", "amperenodes-medium" ] version: widget: select @@ -74,6 +84,7 @@ form: - version - 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 df3693745424a64605d35f8396b7d60dc8bb4d4f..3d253e4da741f0404b33f564bb1e16b82a85e201 100644 --- a/submit.yml.erb +++ b/submit.yml.erb @@ -17,8 +17,8 @@ script: - "--partition=<%= bc_partition %>" - "--time=<%= bc_num_hours.blank? ? 1 : bc_num_hours.to_i %>:00:00" - "--job-name=ood-matlab-<%= version.split("/")[-1] %>" -<%- 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"