submit.yml.erb

<%-
  nodes = bc_num_slots.blank? ? 1 : bc_num_slots.to_i

  case custom_queue
  when "ondemand-p100"
    slurm_args = [
      "--nodes", "#{nodes}",
      "-c", "#{num_cores}",
      "--partition", "ondemand-p100"
    ]
  when "classroomgpu"
    slurm_args = [
      "--nodes", "#{nodes}",
      "-c", "#{num_cores}",
      "--partition", "classroomgpu"
    ]
  when "cryogars"
    slurm_args = [
      "--nodes", "#{nodes}",
      "-c", "#{num_cores}",
      "--partition", "cryogars"
    ]
  else
    slurm_args = [
      "--nodes", "#{nodes}",
      "-c", "#{num_cores}",
      "--partition", "ondemand"
    ]
  end
  if enable_gpu == "1"
    slurm_args += ["--gpus-per-node", "1"]
  end
%>
---
batch_connect:
  template: "basic"
  conn_params:
    - code_server_version
script:
  native:
  <%- slurm_args.each do |arg| %>
    - "<%= arg %>"
  <%- end %>
    - "-J VSCode"