fix form

form.yml.erb

@@ -5,6 +5,8 @@ form:
   - account
   - custom_queue
   - bc_num_hours 
+  - num_nodes
+  - num_tasks
   - num_gpus
   - version
   - working_dir  
@@ -19,6 +21,16 @@ attributes:
       - "<%= 'qgpu' %>"
       - "<%= 'qgpu_exp' %>"
 
+  num_nodes:
+    label: "Nodes"
+    value: "1"
+    required: true
+    
+  num_tasks:
+    label: "Tasks"
+    value: "1"   
+    required: true      
+
   num_gpus:
     label: "GPUs"  
     widget: select

submit.yml.erb

----
+<%- 
+  gpu = custom_queue.include?("gpu")
+-%>
+
 batch_connect:
   template: "basic"
   conn_params:
     - code_server_version  
+
 script:
   native:
     - "-A"
     - "<%= account %>"    
     - "-p"
     - "<%= custom_queue %>"
+    - "-N"
+    - "<%= num_nodes %>"
+    - "-n"
+    - "<%= num_tasks %>"  
+    <%- if gpu -%>
+    - "--gpus <%= num_gpus %>"
+    - "--comment='use:xorg=True'"
+    <%- end -%>    
\ No newline at end of file