vagrant-slurm/Vagrantfile

54 lines
1.4 KiB
Ruby
Raw Permalink Normal View History

# -*- 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
2024-08-13 03:58:25 +00:00
JOIN_TIMEOUT = settings['JOIN_TIMEOUT'] || 120
# 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
# Install and Setup Slurm
config.vm.provision "shell", inline: <<-SHELL
export JOIN_TIMEOUT=#{JOIN_TIMEOUT}
/bin/bash /vagrant/provision.sh
SHELL
end
end
end