diff --git a/form.yml.erb b/form.yml.erb index f6ccb9eea24d75377d55d779991226d780637cab..227d928d94704dc3094f9551d8fd4d703321ba21 100644 --- a/form.yml.erb +++ b/form.yml.erb @@ -19,6 +19,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 @@ -34,6 +41,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 @@ -46,6 +56,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 dd2114ba15437d2d8045ffdc8b888e20b9b06644..2cf61d81dac879526034c5e131390853badcd38f 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-igv" -<%- 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"