nixcfg/shared/slurm.nix

{ pkgs, lib, config, nodes, ... }:

with lib;

{
  services.slurm = {
    clusterName = "mogli";

    controlMachine = "manager";

    nodeName = map
      (node: "${node.config.networking.hostName} CPUs=64")
      (filter
        (node: elem "node" node.config.deployment.tags)
        (attrValues nodes));

    partitionName = [
      "all Nodes=ALL AllowGroups=cluster Default=YES MaxTime=INFINITE State=UP"
    ];

    extraConfig = ''
      MailProg=${pkgs.coreutils}/bin/false
    '';
  };

  services.munge = {
    enable = true;
    password = config.sops.secrets."munge/password".path;
  };

  sops.secrets."munge/password" = {
    sopsFile = ./secrets.yaml;
    owner = "munge";
  };
}