submit.yml.erb

<%- 
  gpu = custom_queue.include?("gpu")
  num_nodes = 1
  num_tasks = 1
  ng = 8
-%>

batch_connect:
  template: "vnc"
script:
  native:
    - "-A"
    - "<%= account %>"    
    - "-p"
    - "<%= custom_queue %>"
    - "-N"
    - "<%= num_nodes %>"
    - "-n"
    - "<%= num_tasks %>"  
    <%- if gpu -%>
    - "--gpus <%= ng %>"
    - "--comment='use:xorg=True'"
    <%- end -%>
    <%- if reservation && reservation != "none" -%>
    - "--reservation=<%= reservation %>"
    <%- end -%>