bc_osc_mathematica/submit.yml.erb at master · OSC/bc_osc_mathematica · 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
<%-

  ppn = num_cores.to_i

  case node_type
  when "any"
      slurm_args = [ "--ntasks-per-node", "#{ppn}" ]
  when "hugemem"
    partition = bc_num_slots.to_i > 1 ? "hugemem-parallel" : "hugemem"
    slurm_args = [ "--ntasks-per-node", "#{ppn}", "--partition", partition ]
  when "gpu"
    slurm_args = [ "--ntasks-per-node", "#{ppn}", "--gpus-per-node", "1" ]
  when "any40-core"
    slurm_args = [ "--ntasks-per-node", "#{ppn}", "--contstraint", "48core" ]
  when "any48-core"
    slurm_args = [ "--ntasks-per-node", "#{ppn}", "--contstraint", "48core" ]
  when "gpu-48core"
    slurm_args = [ "--ntasks-per-node", "#{ppn}", "--gpus-per-node", "1", "--constraint", "48core" ]
  when "gpu-40core"
    slurm_args = [ "--ntasks-per-node", "#{ppn}", "--gpus-per-node", "1", "--constraint", "40core" ]
  else
    slurm_args = [ "--ntasks-per-node", "#{ppn}" ]
  end
%>
---
batch_connect:
  template: vnc
script:
  native:
  <%- slurm_args.each do |arg| %>
    - "<%= arg %>"
  <%- end %>