bc_osc_matlab/submit.yml.erb at main · bsurc/bc_osc_matlab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
<%-
  nodes = bc_num_slots.blank? ? 1 : bc_num_slots.to_i

  case custom_queue
  when "ondemand-p100"
    slurm_args = [
      "--nodes", "#{nodes}",
      "--ntasks-per-node", "14",
      "--gpus-per-node", "1",
      "--partition", "ondemand-p100"
    ]
  when "classroomgpu"
    slurm_args = [
      "--nodes", "#{nodes}",
      "--ntasks-per-node", "14",
      "--gpus-per-node", "1",
      "--partition", "classroomgpu"
    ]
  when "cryogars"
    slurm_args = [
      "--nodes", "#{nodes}",
      "--ntasks-per-node", "32",
      "--gpus-per-node", "1",
      "--partition", "cryogars"
    ]
  else
    slurm_args = [
      "--nodes", "#{nodes}",
      "--ntasks-per-node", "4",
      "--partition", "ondemand"
    ]
  end
%>
---
batch_connect:
  template: vnc
script:
  native:
  <%- slurm_args.each do |arg| %>
    - "<%= arg %>"
  <%- end %>
    - "-J Matlab"