Slurm Workload Manager - Support for Multi-core/Multi-thread Architectures

# Support for Multi-core/Multi-thread Architectures

## Contents

- Definitions
- Overview of srun flags
- Motivation behind high-level srun flags
- Extensions to sinfo/squeue/scontrol
- Configuration settings in slurm.conf

## Definitions

BaseBoard
  Also called motherboard.
LDom
  Locality domain or NUMA domain. May be equivalent to BaseBoard or Socket.
Socket/Core/Thread
  Figure 1 illustrates the notion of Socket, Core and Thread as it is defined in Slurm's multi-core/multi-thread support documentation.
CPU
  Depending upon system configuration, this can be either a core or a thread.

Figure 1: Definitions of Socket, Core, & Thread

Affinity
  The state of being bound to a specific logical processor.
Affinity Mask
  A bitmask where indices correspond to logical processors. The least significant bit corresponds to the first logical processor number on the system, while the most significant bit corresponds to the last logical processor number on the system. A '1' in a given position indicates a process can run on the associated logical processor.
Fat Masks
  Affinity masks with more than 1 bit set allowing a process to run on more than one logical processor.

## Overview of srun flags

Many flags have been defined to allow users to better take advantage of this architecture by explicitly specifying the number of sockets, cores, and threads required by their application. Table 1 summarizes these options.

Low-level (explicit binding)
--cpu-bind=... | Explicit process affinity binding and control options
High-level (automatic mask generation)
--sockets-per-node=S | Number of sockets in a node to dedicate to a job (minimum)
--cores-per-socket=C | Number of cores in a socket to dedicate to a job (minimum)
--threads-per-core=T | Minimum number of threads in a core to dedicate to a job. In task layout, use the specified maximum number of threads per-core.
-B S[:C[:T]] | Combined shortcut option for --sockets-per-node, --cores-per_cpu, --threads-per_core
Task Distribution Options
-m / --distribution | Distributions of: arbitrary | block | cyclic | plane=x | [block|cyclic]:[block|cyclic|fcyclic]
Memory as a consumable resource
--mem=mem | amount of real memory per node required by the job.
--mem-per-cpu=mem | amount of real memory per allocated CPU required by the job.
Task invocation control
--cpus-per-task=CPUs | number of CPUs required per task
| --ntasks-per-node=ntasks | number of tasks to invoke on each node | --ntasks-per-socket=ntasks | number of tasks to invoke on each socket | --ntasks-per-core=ntasks | number of tasks to invoke on each core | --overcommit | Permit more than one task per CPU
Application hints
--hint=compute_bound | use all cores in each socket
| --hint=memory_bound | use only one core in each socket | --hint=[no]multithread | [don't] use extra threads with in-core multi-threading
Resources reserved for system use
--core-spec=cores | Count of cores to reserve for system use
| --thread-spec=threads | Count of threads to reserve for system use

Table 1: srun flags to support the multi-core/multi-threaded environment

It is important to note that many of these flags are only meaningful if the processes have some affinity to specific CPUs and (optionally) memory. Inconsistent options generally result in errors. Task affinity is configured using the TaskPlugin parameter in the slurm.conf file. Several options exist for the TaskPlugin depending upon system architecture and available software, any of them except "task/none" will bind tasks to CPUs. See the "Task Launch" section if generating slurm.conf via configurator.html.

### Low-level --cpu-bind=... - Explicit binding interface

The following srun flag provides a low-level core binding interface:

    --cpu-bind=        Bind tasks to CPUs
        q[uiet]         quietly bind before task runs (default)
        v[erbose]       verbosely report binding before task runs
        no[ne]          don't bind tasks to CPUs (default)
        rank            bind by task rank
        map_cpu:<list>  specify a CPU ID binding for each task
                        where <list> is
                        <cpuid1>,<cpuid2>,...<cpuidN>
        mask_cpu:<list> specify a CPU ID binding mask for each
                        task where <list> is
                        <mask1>,<mask2>,...<maskN>
        rank_ldom       bind task by rank to CPUs in a NUMA
                        locality domain
        map_ldom:<list> specify a NUMA locality domain ID
                        for each task where <list> is
                        <ldom1>,<ldom2>,...<ldomN>
        rank_ldom       bind task by rank to CPUs in a NUMA
                        locality domain where <list> is
                        <ldom1>,<ldom2>,...<ldomN>
        mask_ldom:<list> specify a NUMA locality domain ID mask
                        for each task where <list> is
                        <ldom1>,<ldom2>,...<ldomN>
        ldoms           auto-generated masks bind to NUMA locality
                        domains
        sockets         auto-generated masks bind to sockets
        cores           auto-generated masks bind to cores
        threads         auto-generated masks bind to threads
        help            show this help message

The affinity can be either set to either a specific logical processor (socket, core, threads) or at a coarser granularity than the lowest level of logical processor (core or thread). In the later case the processes are allowed to utilize multiple processors within a specific socket or core.

Examples:

- srun -n 8 -N 4 --cpu-bind=mask_cpu:0x1,0x4 a.out
- srun -n 8 -N 4 --cpu-bind=mask_cpu:0x3,0xD a.out

See also 'srun --cpu-bind=help' and 'man srun'

### High-level -B S[:C[:T]] - Automatic mask generation interface

We have updated the node selection infrastructure with a mechanism that allows selection of logical processors at a finer granularity. Users are able to request a specific number of nodes, sockets, cores, and threads:

    -B --extra-node-info=S[:C[:T]]            Expands to:
      --sockets-per-node=S   number of sockets per node to allocate
      --cores-per-socket=C   number of cores per socket to allocate
      --threads-per-core=T   number of threads per core to allocate
                    each field can be 'min' or wildcard '*'

         Total cpus requested = (Nodes) x (S x C x T)

Examples:

- srun -n 8 -N 4 -B 2:1 a.out
- srun -n 8 -N 4 -B 2 a.out
  note: compare the above with the previous corresponding --cpu-bind=... examples
- srun -n 16 -N 4 a.out
- srun -n 16 -N 4 -B 2:2:1 a.out
- srun -n 16 -N 4 -B 2:2:1 a.out
  or
- srun -n 16 -N 4 --sockets-per-node=2 --cores-per-socket=2 --threads-per-core=1 a.out
- srun -n 16 -N 2-4 -B '1:*:1' a.out
- srun -n 16 -N 4-2 -B '2:*:1' a.out
- srun -n 16 -N 4-4 -B '1:1' a.out

Notes:

- Adding --cpu-bind=no to the command line will cause the processes to not be bound the logical processors.
- Adding --cpu-bind=verbose to the command line (or setting the CPU_BIND environment variable to "verbose") will cause each task to report the affinity mask in use
- Binding is on by default when -B is used. The default binding on multi-core/multi-threaded systems is equivalent to the level of resource enumerated in the -B option.

See also 'srun --help' and 'man srun'

### Task distribution options: Extensions to -m / --distribution

The -m / --distribution option for distributing processes across nodes has been extended to also describe the distribution within the lowest level of logical processors. Available distributions include:
arbitrary | block | cyclic | plane=x | [block|cyclic]:[block|cyclic|fcyclic]

The plane distribution (plane=x) results in a block:cyclic distribution of blocksize equal to x. In the following we use "lowest level of logical processors" to describe sockets, cores or threads depending of the architecture. The distribution divides the cluster into planes (including a number of the lowest level of logical processors on each node) and then schedule first within each plane and then across planes.

For the two dimensional distributions ([block|cyclic]:[block|cyclic|fcyclic]), the second distribution (after ":") allows users to specify a distribution method for processes within a node and applies to the lowest level of logical processors (sockets, core or thread depending on the architecture). When a task requires more than one CPU, the cyclic will allocate all of those CPUs as a group (i.e. within the same socket if possible) while fcyclic would distribute each of those CPU of the in a cyclic fashion across sockets.

The binding is enabled automatically when high level flags are used as long as the task/affinity plug-in is enabled. To disable binding at the job level use --cpu-bind=no.

The distribution flags can be combined with the other switches:

- srun -n 16 -N 4 -B '2:*:1' -m block:cyclic --cpu-bind=socket a.out
- srun -n 16 -N 4 -B '2:*:1' -m plane=2 --cpu-bind=core a.out
- srun -n 16 -N 4 -B '2:*:1' -m plane=2 a.out

The default distribution on multi-core/multi-threaded systems is equivalent to -m block:cyclic with --cpu-bind=thread.

See also 'srun --help'

### Memory as a Consumable Resource

The --mem flag specifies the maximum amount of memory in MB needed by the job per node. This flag is used to support the memory as a consumable resource allocation strategy.

    --mem=MB      maximum amount of real memory per node
                  required by the job.

This flag allows the scheduler to co-allocate jobs on specific nodes given that their added memory requirement do not exceed the total amount of memory on the nodes.

In order to use memory as a consumable resource, the select/cons_tres plugin must be first enabled in slurm.conf:

    SelectType=select/cons_tres     # enable consumable resources
    SelectTypeParameters=CR_Memory  # memory as a consumable resource

Using memory as a consumable resource is typically combined with the CPU, Socket, or Core consumable resources using SelectTypeParameters values of: CR_CPU_Memory, CR_Socket_Memory or CR_Core_Memory

See the "Resource Selection" section if generating slurm.conf via configurator.html.

See also 'srun --help' and 'man srun'

### Task invocation as a function of logical processors

The `--ntasks-per-{node,socket,core}=ntasks` flags allow the user to request that no more than `ntasks` be invoked on each node, socket, or core. This is similar to using `--cpus-per-task=ncpus` but does not require knowledge of the actual number of cpus on each node. In some cases, it is more convenient to be able to request that no more than a specific number of ntasks be invoked on each node, socket, or core. Examples of this include submitting an app where only one "task/rank" should be assigned to each node while allowing the job to utilize all of the parallelism present in the node, or submitting a single setup/cleanup/monitoring job to each node of a pre-existing allocation as one step in a larger job script. This can now be specified via the following flags:

    --ntasks-per-node=n    number of tasks to invoke on each node
    --ntasks-per-socket=n  number of tasks to invoke on each socket
    --ntasks-per-core=n    number of tasks to invoke on each core

For example, given a cluster with nodes containing two sockets, each containing two cores, the following commands illustrate the behavior of these flags:

    % srun -n 4 hostname
    hydra12
    hydra12
    hydra12
    hydra12
    % srun -n 4 --ntasks-per-node=1 hostname
    hydra12
    hydra13
    hydra14
    hydra15
    % srun -n 4 --ntasks-per-node=2 hostname
    hydra12
    hydra12
    hydra13
    hydra13
    % srun -n 4 --ntasks-per-socket=1 hostname
    hydra12
    hydra12
    hydra13
    hydra13
    % srun -n 4 --ntasks-per-core=1 hostname
    hydra12
    hydra12
    hydra12
    hydra12

See also 'srun --help' and 'man srun'

### Application hints

Different applications will have various levels of resource requirements. Some applications tend to be computationally intensive but require little to no inter-process communication. Some applications will be memory bound, saturating the memory bandwidth of a processor before exhausting the computational capabilities. Other applications will be highly communication intensive causing processes to block awaiting messages from other processes. Applications with these different properties tend to run well on a multi-core system given the right mappings.

For computationally intensive applications, all cores in a multi-core system would normally be used. For memory bound applications, only using a single core on each socket will result in the highest per core memory bandwidth. For communication intensive applications, using in-core multi-threading (e.g. hyperthreading, SMT, or TMT) may also improve performance. The following command line flags can be used to communicate these types of application hints to the Slurm multi-core support:

    --hint=             Bind tasks according to application hints
        compute_bound   use all cores in each socket
        memory_bound    use only one core in each socket
        [no]multithread [don't] use extra threads with in-core multi-threading
        help            show this help message

For example, given a cluster with nodes containing two sockets, each containing two cores, the following commands illustrate the behavior of these flags. In the verbose --cpu-bind output, tasks are described as 'hostname, task Global_ID Local_ID [PID]':

    % srun -n 4 --hint=compute_bound --cpu-bind=verbose sleep 1
    cpu-bind=MASK - hydra12, task  0  0 [15425]: mask 0x1 set
    cpu-bind=MASK - hydra12, task  1  1 [15426]: mask 0x4 set
    cpu-bind=MASK - hydra12, task  2  2 [15427]: mask 0x2 set
    cpu-bind=MASK - hydra12, task  3  3 [15428]: mask 0x8 set

    % srun -n 4 --hint=memory_bound --cpu-bind=verbose sleep 1
    cpu-bind=MASK - hydra12, task  0  0 [15550]: mask 0x1 set
    cpu-bind=MASK - hydra12, task  1  1 [15551]: mask 0x4 set
    cpu-bind=MASK - hydra13, task  2  0 [14974]: mask 0x1 set
    cpu-bind=MASK - hydra13, task  3  1 [14975]: mask 0x4 set

See also 'srun --hint=help' and 'man srun'

## Motivation behind high-level srun flags

The motivation behind allowing users to use higher level srun flags instead of --cpu-bind is that the later can be difficult to use. The proposed high-level flags are easier to use than --cpu-bind because:

- Affinity mask generation happens automatically when using the high-level flags.
- The length and complexity of the --cpu-bind flag vs. the length of the combination of -B and --distribution flags make the high-level flags much easier to use.

Also as illustrated in the example below it is much simpler to specify a different layout using the high-level flags since users do not have to recalculate mask or CPU IDs. This approach is much simpler than rearranging the mask or map.

Given a 32-process job and a four node, dual-socket, dual-core cluster, we want to use a block distribution across the four nodes and then a cyclic distribution within the node across the physical processors. Below we show how to obtain the wanted layout using 1) high-level flags and 2) --cpubind

### High-Level flags

Using Slurm's high-level flag, users can obtain the above layout with either of the following submissions since --distribution=block:cyclic is the default distribution method.

    $ srun -n 32 -N 4 -B 4:2 --distribution=block:cyclic a.out

or

    $ srun -n 32 -N 4 -B 4:2 a.out

The cores are shown as c0 and c1 and the processors are shown as p0 through p3. The resulting task IDs are:

| c0 | c1
p0 | 0 | 4
p2 | 2 | 6
| c0 | c1
p1 | 1 | 5
p3 | 3 | 7

The computation and assignment of the task IDs is transparent to the user. Users don't have to worry about the core numbering (Section Pinning processes to cores) or any setting any CPU affinities. By default CPU affinity will be set when using multi-core supporting flags.

### Low-level flag --cpu-bind

Using Slurm's --cpu-bind flag, users must compute the CPU IDs or masks as well as make sure they understand the core numbering on their system. Another problem arises when core numbering is not the same on all nodes. The --cpu-bind option only allows users to specify a single mask for all the nodes. Using Slurm high-level flags remove this limitation since Slurm will correctly generate the appropriate masks for each requested nodes.

### On a four dual-socket dual-core node cluster with core block numbering

The cores are shown as c0 and c1 and the processors are shown as p0 through p3. The CPU IDs within a node in the block numbering are: (this information is available from the /proc/cpuinfo file on the system)

| c0 | c1
p0 | 0 | 1
p2 | 4 | 5
| c0 | c1
p1 | 2 | 3
p3 | 6 | 7

resulting in the following mapping for processor/cores and task IDs which users need to calculate:

mapping for processors/cores

| c0 | c1
p0 | 0x01 | 0x02
p2 | 0x10 | 0x20
| c0 | c1
p1 | 0x04 | 0x08
p3 | 0x40 | 0x80

task IDs

| c0 | c1
p0 | 0 | 4
p2 | 2 | 6
| c0 | c1
p1 | 1 | 5
p3 | 3 | 7

The above maps and task IDs can be translated into the following command:

    $ srun -n 32 -N 4 --cpu-bind=mask_cpu:1,4,10,40,2,8,20,80 a.out

or

    $ srun -n 32 -N 4 --cpu-bind=map_cpu:0,2,4,6,1,3,5,7 a.out

### Same cluster but with its core numbered cyclic instead of block

On a system with cyclically numbered cores, the correct mask argument to the srun command looks like: (this will achieve the same layout as the command above on a system with core block numbering.)

    $ srun -n 32 -N 4 --cpu-bind=map_cpu:0,1,2,3,4,5,6,7 a.out

### Block map_cpu on a system with cyclic core numbering

If users do not check their system's core numbering before specifying the map_cpu list and thereby do not realize that the system has cyclic core numbering instead of block numbering then they will not get the expected layout. For example, if they decide to re-use their command from above:

    $ srun -n 32 -N 4 --cpu-bind=map_cpu:0,2,4,6,1,3,5,7 a.out

they get the following unintentional task ID layout:

| c0 | c1
p0 | 0 | 2
p2 | 1 | 3
| c0 | c1
p1 | 4 | 6
p3 | 5 | 7

since the processor IDs within a node in the cyclic numbering are:

| c0 | c1
p0 | 0 | 4
p2 | 2 | 6
| c0 | c1
p1 | 1 | 5
p3 | 3 | 7

The important conclusion is that using the --cpu-bind flag is not trivial and that it assumes that users are experts.

## Extensions to sinfo/squeue/scontrol

Several extensions have also been made to the other Slurm utilities to make working with multi-core/multi-threaded systems easier.

### sinfo

The long version (-l) of the sinfo node listing (-N) has been extended to display the sockets, cores, and threads present for each node. For example:

    % sinfo -N
    NODELIST     NODES PARTITION STATE
    hydra[12-15]     4    parts* idle

    % sinfo -lN
    Thu Sep 14 17:47:13 2006
    NODELIST     NODES PARTITION       STATE CPUS    S:C:T MEMORY TMP_DISK WEIGHT FEATURES REASON
    hydra[12-15]     4    parts*        idle   8+ 2+:4+:1+   2007    41447      1   (null) none

    % sinfo -lNe
    Thu Sep 14 17:47:18 2006
    NODELIST     NODES PARTITION       STATE CPUS    S:C:T MEMORY TMP_DISK WEIGHT FEATURES REASON

    hydra[12-14]     3    parts*        idle    8    2:4:1   2007    41447      1   (null) none
    hydra15          1    parts*        idle   64    8:4:2   2007    41447      1   (null) none

For user specified output formats (-o/--format) and sorting (-S/--sort), the following identifiers are available:

    %X  Number of sockets per node
    %Y  Number of cores per socket
    %Z  Number of threads per core
    %z  Extended processor information: number of
        sockets, core, threads (S:C:T) per node

For example:

    % sinfo -o '%9P %4c %8z %8X %8Y %8Z'
    PARTITION CPUS S:C:T    SOCKETS  CORES    THREADS
    parts*    4    2:2:1    2        2        1

See also 'sinfo --help' and 'man sinfo'

### squeue

For user specified output formats (-o/--format) and sorting (-S/--sort), the following identifiers are available:

    %m  Size of memory (in MB) requested by the job
    %H  Number of requested sockets per node
    %I  Number of requested cores per socket
    %J  Number of requested threads per core
    %z  Extended processor information: number of requested
        sockets, cores, threads (S:C:T) per node

Below is an example squeue output after running 7 copies of:

% srun -n 4 -B 2:2:1 --mem=1024 sleep 100 &

    % squeue -o '%.5i %.2t %.4M %.5D %7H %6I %7J %6z %R'
    JOBID ST TIME NODES SOCKETS CORES THREADS S:C:T NODELIST(REASON)
       17 PD 0:00     1 2       2     1       2:2:1 (Resources)
       18 PD 0:00     1 2       2     1       2:2:1 (Resources)
       19 PD 0:00     1 2       2     1       2:2:1 (Resources)
       13  R 1:27     1 2       2     1       2:2:1 hydra12
       14  R 1:26     1 2       2     1       2:2:1 hydra13
       15  R 1:26     1 2       2     1       2:2:1 hydra14
       16  R 1:26     1 2       2     1       2:2:1 hydra15

The squeue command can also display the memory size of jobs, for example:

    % sbatch --mem=123 tmp
    Submitted batch job 24

    $ squeue -o "%.5i %.2t %.4M %.5D %m"
    JOBID ST TIME NODES MIN_MEMORY
      24   R 0:05     1 123

See also 'squeue --help' and 'man squeue'

### scontrol

The following job settings can be adjusted using scontrol:

    Requested Allocation:
      ReqSockets=<count>  Set the job's count of required sockets
      ReqCores=<count>    Set the job's count of required cores
      ReqThreads=<count>  Set the job's count of required threads

For example:

    # scontrol update JobID=17 ReqThreads=2
    # scontrol update JobID=18 ReqCores=4
    # scontrol update JobID=19 ReqSockets=8

    % squeue -o '%.5i %.2t %.4M %.5D %9c %7H %6I %8J'
    JOBID ST TIME NODES MIN_PROCS SOCKETS CORES THREADS
       17 PD 0:00     1 1         4       2     1
       18 PD 0:00     1 1         8       4     2
       19 PD 0:00     1 1         4       2     1
       13  R 1:35     1 0         0       0     0
       14  R 1:34     1 0         0       0     0
       15  R 1:34     1 0         0       0     0
       16  R 1:34     1 0         0       0     0

The 'scontrol show job' command can be used to display the number of allocated CPUs per node as well as the socket, cores, and threads specified in the request and constraints.

    % srun -N 2 -B 2:1 sleep 100 &
    % scontrol show job 20
    JobId=20 UserId=(30352) GroupId=users(1051)
       Name=sleep
       Priority=4294901749 Partition=parts BatchFlag=0
       AllocNode:Sid=hydra16:3892 TimeLimit=UNLIMITED
       JobState=RUNNING StartTime=09/25-17:17:30 EndTime=NONE
       NodeList=hydra[12-14] NodeListIndices=0,2,-1
       AllocCPUs=1,2,1
       NumCPUs=4 ReqNodes=2 ReqS:C:T=2:1:*
       OverSubscribe=0 Contiguous=0 CPUs/task=0
       MinCPUs=0 MinMemory=0 MinTmpDisk=0 Features=(null)
       Dependency=0 Account=(null) Reason=None Network=(null)
       ReqNodeList=(null) ReqNodeListIndices=-1
       ExcNodeList=(null) ExcNodeListIndices=-1
       SubmitTime=09/25-17:17:30 SuspendTime=None PreSusTime=0

See also 'scontrol --help' and 'man scontrol'

## Configuration settings in slurm.conf

Several slurm.conf settings are available to control the multi-core features described above.

In addition to the description below, also see the "Task Launch" and "Resource Selection" sections if generating slurm.conf via configurator.html.

As previously mentioned, in order for the affinity to be set, the task/affinity plugin must be first enabled in slurm.conf:

    TaskPlugin=task/affinity          # enable task affinity

This setting is part of the task launch specific parameters:

    # o Define task launch specific parameters
    #
    #    "TaskProlog" : Define a program to be executed as the user before each
    #                   task begins execution.
    #    "TaskEpilog" : Define a program to be executed as the user after each
    #                   task terminates.
    #    "TaskPlugin" : Define a task launch plugin. This may be used to
    #                   provide resource management within a node (e.g. pinning
    #                   tasks to specific processors). Permissible values are:
    #      "task/affinity" : CPU affinity support
    #      "task/cgroup"   : bind tasks to resources using Linux cgroup
    #      "task/none"     : no task launch actions, the default
    #
    # Example:
    #
    # TaskProlog=/usr/local/slurm/etc/task_prolog # default is none
    # TaskEpilog=/usr/local/slurm/etc/task_epilog # default is none
    # TaskPlugin=task/affinity                    # default is task/none

Declare the node hardware configuration in slurm.conf:

    NodeName=dualcore[01-16] CoresPerSocket=2 ThreadsPerCore=1

For a more complete description of the various node configuration options see the slurm.conf man page.

Last modified 19 May 2023