pyrox.dev
slurm: impl basic configuration
+50
-19
nixos/modules/services/computing/slurm/slurm.nix
+50
-19
nixos/modules/services/computing/slurm/slurm.nix
···
8
8
# configuration file can be generated by http://slurm.schedmd.com/configurator.html
9
9
configFile = pkgs.writeText "slurm.conf"
10
10
''
11
-
${cfg.extraConfig}
11
+
${optionalString (cfg.controlMachine != null) ''controlMachine=${cfg.controlMachine}''}
12
+
${optionalString (cfg.controlAddr != null) ''controlAddr=${cfg.controlAddr}''}
13
+
${optionalString (cfg.nodeName != null) ''nodeName=${cfg.nodeName}''}
14
+
${optionalString (cfg.partitionName != null) ''partitionName=${cfg.partitionName}''}
15
+
${cfg.extraConfig}
12
16
'';
13
17
in
14
18
···
21
25
services.slurm = {
22
26
23
27
server = {
24
-
enable = mkOption {
25
-
default = false;
26
-
type = types.bool;
27
-
description = ''
28
-
Whether to enable slurm control daemon.
29
-
'';
30
-
};
28
+
enable = mkEnableOption "slurm control daemon";
31
29
32
30
};
33
31
34
32
client = {
35
-
enable = mkOption {
36
-
default = false;
37
-
type = types.bool;
38
-
description = ''
39
-
Whether to enable slurm client daemon.
40
-
'';
41
-
};
33
+
enable = mkEnableOption "slurm rlient daemon";
34
+
35
+
};
36
+
37
+
controlMachine = mkOption {
38
+
type = types.nullOr types.str;
39
+
default = null;
40
+
example = null;
41
+
description = ''
42
+
The short hostname of the machine where SLURM control functions are
43
+
executed (i.e. the name returned by the command "hostname -s", use "tux001"
44
+
rather than "tux001.my.com").
45
+
'';
46
+
};
47
+
48
+
controlAddr = mkOption {
49
+
type = types.nullOr types.str;
50
+
default = cfg.controlMachine;
51
+
example = null;
52
+
description = ''
53
+
Name that ControlMachine should be referred to in establishing a
54
+
communications path.
55
+
'';
56
+
};
42
57
58
+
nodeName = mkOption {
59
+
type = types.nullOr types.str;
60
+
default = null;
61
+
example = "linux[1-32] CPUs=1 State=UNKNOWN";
62
+
description = ''
63
+
Name that SLURM uses to refer to a node (or base partition for BlueGene
64
+
systems). Typically this would be the string that "/bin/hostname -s"
65
+
returns. Note that now you have to write node's parameters after the name.
66
+
'';
67
+
};
68
+
69
+
partitionName = mkOption {
70
+
type = types.nullOr types.str;
71
+
default = null;
72
+
example = "debug Nodes=linux[1-32] Default=YES MaxTime=INFINITE State=UP";
73
+
description = ''
74
+
Name by which the partition may be referenced. Note that now you have
75
+
to write patrition's parameters after the name.
76
+
'';
43
77
};
44
78
45
79
extraConfig = mkOption {
···
50
84
the end of the slurm configuration file.
51
85
'';
52
86
};
53
-
54
87
};
55
88
56
89
};
···
64
97
65
98
systemd.services.slurmd = mkIf (cfg.client.enable) {
66
99
path = with pkgs; [ slurm-llnl coreutils ];
67
-
100
+
68
101
wantedBy = [ "multi-user.target" ];
69
102
after = [ "systemd-tmpfiles-clean.service" ];
70
103
···
73
106
ExecStart = "${pkgs.slurm-llnl}/bin/slurmd -f ${configFile}";
74
107
PIDFile = "/run/slurmd.pid";
75
108
ExecReload = "${pkgs.coreutils}/bin/kill -HUP $MAINPID";
76
-
ExecStop = "${pkgs.coreutils}/bin/kill $MAINPID";
77
109
};
78
110
};
79
111
···
89
121
ExecStart = "${pkgs.slurm-llnl}/bin/slurmctld";
90
122
PIDFile = "/run/slurmctld.pid";
91
123
ExecReload = "${pkgs.coreutils}/bin/kill -HUP $MAINPID";
92
-
ExecStop = "${pkgs.coreutils}/bin/kill $MAINPID";
93
124
};
94
125
environment = { SLURM_CONF = "${configFile}"; };
95
126
};