The Slurm Scheduler

Introduction

Running software on the JADE system is accomplished via batch jobs, i.e. in an unattended, non-interactive manner. Typically a user logs in to the JADE login nodes, prepares a job script and submits it to the job queue.

Jobs on JADE are managed by the Slurm batch system , which is in charge of:

  • allocating the computer resources requested for the job,
  • running the job and
  • reporting the outcome of the execution back to the user.

Running a job involves, at the minimum, the following steps

  • preparing a submission script and
  • submitting the job to execution.

This guide describes basic job submission and monitoring for Slurm. The generic topics in the guide are:

  • the main Slurm commands,
  • preparing a submission script,
  • submitting a job to the queue,
  • monitoring a job execution,
  • deleting a job from the queue and
  • important environment variables.

Additionally, the following topics specific to JADE are covered (under construction)

  • slurm partitions,
  • using fast local storage and
  • controlling affinity.

Commands

The table below gives a short description of the most used Slurm commands.

Command Description
sacct report job accounting information about active or completed jobs
salloc allocate resources for a job in real time (typically used to allocate resources and spawn a shell, in which the srun command is used to launch parallel tasks)
sbatch submit a job script for later execution (the script typically contains one or more srun commands to launch parallel tasks)
scancel cancel a pending or running job
sinfo reports the state of partitions and nodes managed by Slurm (it has a variety of filtering, sorting, and formatting options)
squeue reports the state of jobs (it has a variety of filtering, sorting, and formatting options), by default, reports the running jobs in priority order followed by the pending jobs in priority order
srun used to submit a job for execution in real time

All Slurm commands have extensive help through their man pages e.g.:

man sbatch

shows you the help pages for the sbatch command.

Preparing a submission script

A submission script is a Linux shell script that

  • describes the processing to carry out (e.g. the application, its input and output, etc.) and
  • requests computer resources (number of GPUs, amount of memory, etc.) to use for processing.

The simplest case is that of a job that requires a single node with the following requirements:

  • the job uses 1 node,
  • the application is a single process,
  • the job uses a single GPU,
  • the job will run for no more than 10 hours,
  • the job is given the name “job123” and
  • the user should be emailed when the job starts and stops or aborts.

Supposing the application run is called myCode and takes no command line arguments, the following submission script runs the application in a single job::

#!/bin/bash

# set the number of nodes
#SBATCH --nodes=1

# set max wallclock time
#SBATCH --time=10:00:00

# set name of job
#SBATCH --job-name=job123

# set number of GPUs
#SBATCH --gres=gpu:1

# mail alert at start, end and abortion of execution
#SBATCH --mail-type=ALL

# send mail to this address
#SBATCH --mail-user=john.brown@gmail.com

# run the application
myCode

The script starts with #!/bin/bash (also called a shebang), which makes the submission script a Linux bash script.

The script continues with a series of lines starting with #, which represent bash script comments. For Slurm, the lines starting with #SBATCH are directives that request job scheduling resources. (Note: it is important that you put all the directives at the top of a script, before any other commands; any #SBATCH directive coming after a bash script command is ignored!)

The resource request #SBATCH --nodes=n determines how many compute nodes a job are allocated by the scheduler; only 1 node is allocated for this job.

The maximum walltime is specified by #SBATCH --time=T, where T has format H:M:S. Normally, a job is expected to finish before the specified maximum walltime. After the walltime reaches the maximum, the job terminates regardless whether the job processes are still running or not.

The name the job is identified by in the queue can be specified too with #SBATCH --job-name=name.

Lastly, an email notification is sent if an address is specified with #SBATCH --mail-user=<email_address>. The notification options can be set with #SBATCH --mail-type=<type>, where <type> may be BEGIN, END, FAIL, REQUEUE or ALL (for any change of job state).

The final part of a script is normal Linux bash script and describes the set of operations to follow as part of the job. The job starts in the same folder where it was submitted (unless an alternative path is specified), and with the same environment variables (modules, etc.) that the user had at the time of the submission. In this example, this final part only involves invoking the myCode application executable.

Submitting jobs with the command sbatch

Once you have a submission script ready (e.g called submit.sh), the job is submitted to the execution queue with the command sbatch script.sh. The queueing system prints a number (the job id) almost immediately and returns control to the linux prompt. At this point the job is in the submission queue.

Once the job submitted, it will sit in a pending state until the resources have been allocated to your job (the length of time your job is in the pending state is dependent upon a number of factors including how busy the system is and what resources you are requesting). You can monitor the progress of the job using the command squeue (see below).

Once the job starts to run you will see files with names such as slurm-1234.out either in the directory you submitted the job from (default behaviour) or in the directory where the script was instructed explicitly to change to.

Monitoring jobs with the command squeue

squeue is the main command for monitoring the state of systems, groups of jobs or individual jobs.

The command squeue prints the list of current jobs. The list looks something like this:

| JOBID PARTITION     NAME     USER ST       TIME  NODES NODELIST(REASON)
|  2497     devel     srun      bob  R       0:07      1 dgj416
|  2499       big     test1    mary  R       0:22      4 dgj[101,104]
|  2511     small     test2   steve PD       0:00      4 (Resources)

The first column gives the job ID, the second the partition where the job was submitted, the third the name of the job (specified by the user in the submission script) and the fourth the user ID of the job owner. The fifth is the status of the job (R = running, PD = pending, CA = cancelled, CF = configuring, CG = completing, CD = completed, F = failed). The sixth column gives the elapsed time for each particular job. Finally, there are the number of nodes requested and the nodelist where the job is running (or the cause that it is not running).

Some useful command line options for squeue include:

  • -u for showing the status of all the jobs of a particular user, e.g. squeue -u bob;
  • -l for showing more of the available information;
  • -j for showing information regarding a particular job ID, e.g. squeue -j 7890;
  • --start to report the expected start time of pending jobs.

Read all the options for squeue on the Linux manual using the command man squeue, including how to personalize the information to be displayed.

Deleting jobs with the command scancel

Use the scancel command to delete a job, e.g. scancel 1121 to delete job with ID 1121. Any user can delete their own jobs at any time, whether the job is pending (waiting in the queue) or running. A user cannot delete the jobs of another user. Normally, there is a (small) delay between the execution of the scancel command and the time when the job is dequeued and killed.

Environment variables

At the time a job is launched into execution, Slurm defines multiple environment variables, which can be used from within the submission script to define the correct workflow of the job. A few useful environment variables are the following:

  • SLURM_SUBMIT_DIR, which points to the directory where the sbatch command is issued;
  • SLURM_JOB_NODELIST, which returns the list of nodes allocated to the job;
  • SLURM_JOB_ID, which is a unique number Slurm assigns to a job.

In most cases, SLURM_SUBMIT_DIR does not have to be used, as the job lands by default in the directory where the Slurm command sbatch was issued.

SLURM_SUBMIT_DIR can be useful in a submission script when files must be copied to/from a specific directory that is different from the directory where sbatch was issued.

SLURM_JOB_ID is useful to tag job specific files and directories (typically output files or run directories) in order to identify them as produced by a particular job. For instance, the submission script line

myApp &> $SLURM_JOB_ID.out

runs the application myApp and redirects the standard output (and error) to a file whose name is given by the job ID. Note: the job ID is a number assigned by Slurm and differs from the character string name given to the job in the submission script by the user.