Using node-local storage: Difference between revisions

From Alliance Doc
Jump to navigation Jump to search
No edit summary
No edit summary
Line 32: Line 32:
<!--T:6-->
<!--T:6-->
This may not work if the input is too large, or if it must be read by processes on different nodes.
This may not work if the input is too large, or if it must be read by processes on different nodes.
See <i>Amount of space</i> and <i>Multi-node jobs</i> below for more.
See <i>Amount of space</i> and <i>Multinode jobs</i> below for more.


== Executable files and libraries == <!--T:7-->
== Executable files and libraries == <!--T:7-->
Line 56: Line 56:
job ends.  If a job times out, then the last few lines of the job script might not  
job ends.  If a job times out, then the last few lines of the job script might not  
be executed.  This can be addressed two ways:
be executed.  This can be addressed two ways:
* First, obviously, request enough run time to let the application finish.  We understand that this isn't always possible.
* First, obviously, request enough runtime to let the application finish, although we understand that this isn't always possible.
* Write [[Points_de_contrôle/en|checkpoints]] to network storage, not to <code>$SLURM_TMPDIR</code>.
* Write [[Points_de_contrôle/en|checkpoints]] to network storage, not to <code>$SLURM_TMPDIR</code>.


Line 62: Line 62:


<!--T:28-->
<!--T:28-->
You can use [https://slurm.schedmd.com/sbatch.html <code>--signal</code>] to get Slurm to send your script a signal shortly before the run-time expires.   
You can use [https://slurm.schedmd.com/sbatch.html <code>--signal</code>] to get Slurm to send your script a signal shortly before the runtime expires.   
To take advantage of this, write a shell function which copies your output from <code>$SLURM_TMPDIR</code> back to network storage,  
To take advantage of this, write a shell function which copies your output from <code>$SLURM_TMPDIR</code> back to network storage,  
and use the <code>trap</code> shell command to associate the function with the signal.
and use the <code>trap</code> shell command to associate the function with the signal.
This may be useful if your run-time estimate is uncertain,
This may be useful if your runtime estimate is uncertain,
or if you are chaining together several Slurm jobs to complete a long calculation.
or if you are chaining together several Slurm jobs to complete a long calculation.
However, it will not preserve the contents of <code>$SLURM_TMPDIR</code> in the case of a node failure.
However, it will not preserve the contents of <code>$SLURM_TMPDIR</code> in the case of a node failure.
Line 72: Line 72:
for an example script and detailed guidance.
for an example script and detailed guidance.


= Multi-node jobs = <!--T:12-->
= Multinode jobs = <!--T:12-->


<!--T:13-->
<!--T:13-->

Revision as of 17:39, 15 March 2023

Other languages:


When Slurm starts a job, it creates a temporary directory on each node assigned to the job. It then sets the full path name of that directory in an environment variable called SLURM_TMPDIR.

Because this directory resides on local disk, input and output (I/O) to it is almost always faster than I/O to a network storage (/project, /scratch, or /home). Specifically, local disk is better for frequent small I/O transactions than network storage. Any job doing a lot of input and output (which is most jobs!) may expect to run more quickly if it uses $SLURM_TMPDIR instead of network storage.

The temporary character of $SLURM_TMPDIR makes it more trouble to use than network storage. Input must be copied from network storage to $SLURM_TMPDIR before it can be read, and output must be copied from $SLURM_TMPDIR back to network storage before the job ends to preserve it for later use.

Input

In order to read data from $SLURM_TMPDIR, you must first copy the data there. In the simplest case you can do this with cp or rsync:

cp /project/def-someone/you/input.files.* $SLURM_TMPDIR/

This may not work if the input is too large, or if it must be read by processes on different nodes. See Amount of space and Multinode jobs below for more.

Executable files and libraries

A special case of input is the application code itself. In order to run the application, the shell started by Slurm must open at least an application file, which it typically reads from network storage. But few applications these days consist of exactly one file; most also need several other files (such as libraries) in order to work.

We particularly find that using an application in a Python virtual environment generates a large number of small I/O transactions—more than it takes to create the virtual environment in the first place. This is why we recommend creating virtual environments inside your jobs using $SLURM_TMPDIR.

Output

Output data must be copied from $SLURM_TMPDIR back to some permanent storage before the job ends. If a job times out, then the last few lines of the job script might not be executed. This can be addressed two ways:

  • First, obviously, request enough runtime to let the application finish, although we understand that this isn't always possible.
  • Write checkpoints to network storage, not to $SLURM_TMPDIR.

Signal trapping

You can use --signal to get Slurm to send your script a signal shortly before the runtime expires. To take advantage of this, write a shell function which copies your output from $SLURM_TMPDIR back to network storage, and use the trap shell command to associate the function with the signal. This may be useful if your runtime estimate is uncertain, or if you are chaining together several Slurm jobs to complete a long calculation. However, it will not preserve the contents of $SLURM_TMPDIR in the case of a node failure. See this page from le Centre Régional Informatique et d'Applications Numériques de Normandie (CRIANN) for an example script and detailed guidance.

Multinode jobs

If a job spans multiple nodes and some data is needed on every node, then a simple cp or tar -x will not suffice.

Copy files

Copy one or more files to the SLURM_TMPDIR directory on every node allocated like this:

Question.png
[name@server ~]$ srun --ntasks=$SLURM_NNODES --ntasks-per-node=1 cp file [files...] $SLURM_TMPDIR

Compressed archives

ZIP

Extract to the SLURM_TMPDIR:

Question.png
[name@server ~]$ srun --ntasks=$SLURM_NNODES --ntasks-per-node=1 unzip archive.zip -d $SLURM_TMPDIR

Tarball

Extract to the SLURM_TMPDIR:

Question.png
[name@server ~]$ srun --ntasks=$SLURM_NNODES --ntasks-per-node=1 tar -xvf archive.tar.gz -C $SLURM_TMPDIR

Amount of space

At Niagara $SLURM_TMPDIR is implemented as "RAMdisk", so the amount of space available is limited by the memory on the node, less the amount of RAM used by your application. See Data management at Niagara for more.

At the general-purpose clusters the amount of space available depends on the cluster and the node to which your job is assigned.

cluster space in $SLURM_TMPDIR size of disk
Béluga 370G 480G
Cedar 840G 960G
Graham 750G 960G
Narval 800G 960G

The table above gives the amount of space in $SLURM_TMPDIR on the smallest node in each cluster. If your job reserves whole nodes then you can reasonably assume that this much space is available to you in $SLURM_TMPDIR on each node. However, if the job requests less than a whole node, then other jobs may also write to the same filesystem (but not the same directory!), reducing the space available to your job.

Some nodes at each site have more local disk than shown above. See "Node characteristics" at the appropriate cluster's page (Béluga, Cedar, Graham, Narval) for guidance.