Kris Lamoureux
9a402948c8
- Fix MUNGE key generation and distribution - Slurm config using easy version of the config tool - Ensure JOIN_TIMEOUT is passed to the provisioning script - Execute hostname command as Slurm job for verification
60 lines
1.6 KiB
Ruby
60 lines
1.6 KiB
Ruby
# -*- mode: ruby -*-
|
|
# vi: set ft=ruby :
|
|
|
|
# Load override settings
|
|
require 'yaml'
|
|
settings_path = '.settings.yml'
|
|
settings = {}
|
|
|
|
if File.exist?(settings_path)
|
|
settings = YAML.load_file(settings_path)
|
|
end
|
|
|
|
# Default Vagrant settings
|
|
VAGRANT_BOX = settings['VAGRANT_BOX'] || 'debian/bookworm64'
|
|
VAGRANT_CPU = settings['VAGRANT_CPU'] || 2
|
|
VAGRANT_MEM = settings['VAGRANT_MEM'] || 2048
|
|
SSH_FORWARD = settings['SSH_FORWARD'] || false
|
|
|
|
# Default Slurm settings
|
|
SLURM_NODES = settings['SLURM_NODES'] || 4
|
|
JOIN_TIMEOUT = settings['JOIN_TIMEOUT'] || 60
|
|
|
|
# Node settings overrides
|
|
if File.exist?('nodes.rb')
|
|
require_relative 'nodes.rb'
|
|
else
|
|
# Using all defaults
|
|
NODES = {}
|
|
end
|
|
|
|
HOSTS = Array(1..SLURM_NODES)
|
|
Vagrant.configure(2) do |vm_config|
|
|
HOSTS.each do |count|
|
|
vm_config.vm.define "node#{count}" do |config|
|
|
config.vm.hostname = "node#{count}"
|
|
config.vm.box = NODES.dig("node#{count}", 'BOX') || VAGRANT_BOX
|
|
config.ssh.forward_agent =
|
|
NODES.dig("node#{count}", 'SSH') || SSH_FORWARD
|
|
|
|
# Libvirt
|
|
config.vm.provider :libvirt do |virt|
|
|
virt.memory = NODES.dig("node#{count}", 'MEM') || VAGRANT_MEM
|
|
virt.cpus = NODES.dig("node#{count}", 'CPU') || VAGRANT_CPU
|
|
end
|
|
|
|
# VirtualBox
|
|
config.vm.provider :virtualbox do |vbox|
|
|
vbox.memory = NODES.dig("node#{count}", 'MEM') || VAGRANT_MEM
|
|
vbox.cpus = NODES.dig("node#{count}", 'CPU') || VAGRANT_CPU
|
|
end
|
|
|
|
# Install and Setup Slurm
|
|
config.vm.provision "shell", inline: <<-SHELL
|
|
export JOIN_TIMEOUT=#{JOIN_TIMEOUT}
|
|
/bin/bash /vagrant/provision.sh
|
|
SHELL
|
|
end
|
|
end
|
|
end
|