From d7bba2316aa59854710f83617de925cf924a601e Mon Sep 17 00:00:00 2001
From: Ravi Tripathi <ravi89@uab.edu>
Date: Tue, 13 Jun 2023 14:42:36 -0500
Subject: [PATCH] Adding amperenodes partition to Ansys 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 f2ef4c0..f3c441a 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 df36937..3d253e4 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"
-- 
GitLab