Hyper-Q / MPS: Difference between revisions
No edit summary |
No edit summary |
||
Line 41: | Line 41: | ||
<!--T:12--> | <!--T:12--> | ||
{{File| | {{File|name=script.sh | ||
|contents= | |||
#!/bin/bash | #!/bin/bash | ||
#SBATCH --gpus-per-node=v100:1 | |||
#SBATCH --time=0-10:00 | |||
#SBATCH --mem-per-cpu=8G | |||
#SBATCH --cpus-per-task=8 | |||
mkdir -p $HOME/tmp | |||
export CUDA_MPS_LOG_DIRECTORY=$HOME/tmp | |||
nvidia-cuda-mps-control -d | |||
for ((i=0; i<8; i++)) | |||
do | |||
echo $i | |||
./my_code $i & | |||
done | |||
wait | |||
}} | }} | ||
Revision as of 19:15, 1 December 2023
Overview[edit]
Hyper-Q (or MPS) is a feature of NVIDIA GPUs. It is available in GPUs with CUDA compute capability 3.5 and higher,[1] which is all GPUs currently deployed on Alliance general-purpose clusters (Béluga, Cedar, Graham, and Narval).
- The MPS runtime architecture is designed to transparently enable co-operative multi-process CUDA applications, typically MPI jobs, to utilize Hyper-Q capabilities on the latest NVIDIA (Kepler and later) GPUs. Hyper-Q allows CUDA kernels to be processed concurrently on the same GPU; this can benefit performance when the GPU compute capacity is underutilized by a single application process.
In our tests, MPS may increase the total GPU flop rate even when the GPU is being shared by unrelated CPU processes. That means that MPS is great for CUDA applications with relatively small problem sizes, which on their own cannot efficiently saturate modern GPUs with thousands of cores.
MPS is not enabled by default, but it is straightforward to do. Execute the following commands before running your CUDA application:
[name@server ~]$ export CUDA_MPS_PIPE_DIRECTORY=/tmp/nvidia-mps
[name@server ~]$ export CUDA_MPS_LOG_DIRECTORY=/tmp/nvidia-log
[name@server ~]$ nvidia-cuda-mps-control -d
Then you can use the MPS feature if you have more than one CPU thread accessing the GPU. This will happen if you run a hybrid MPI/CUDA application, a hybrid OpenMP/CUDA application, or multiple instances of a serial CUDA application (GPU farming).
Additional details on MPS can be found here: CUDA Multi Process Service (MPS) - NVIDIA Documentation.
GPU farming[edit]
One situation when the MPS feature can be very useful is when you need to run multiple instances of a CUDA application, but the application is too small to saturate a modern GPU. MPS allows you to run multiple instances of the application sharing a single GPU, as long as there is enough of GPU memory for all of the instances of the application. In many cases this should result in a significantly increased throughput from all of your GPU processes.
Here is an example of a job script to set up GPU farming:
#!/bin/bash
#SBATCH --gpus-per-node=v100:1
#SBATCH --time=0-10:00
#SBATCH --mem-per-cpu=8G
#SBATCH --cpus-per-task=8
mkdir -p $HOME/tmp
export CUDA_MPS_LOG_DIRECTORY=$HOME/tmp
nvidia-cuda-mps-control -d
for ((i=0; i<8; i++))
do
echo $i
./my_code $i &
done
wait
In the above example, we share a single V100 GPU between 8 instances of my_code
(which takes a single argument-- the loop index $i). We request 8 CPU cores (#SBATCH -c 8) so there is one CPU core per application instance. The two important elements are &
on the code execution line, which sends the code processes to the background, and the wait
command at the end of the script, which ensures that the job runs until all background processes end.
- ↑ For a table relating NVIDIA GPU model names, architecture names, and CUDA compute capabilities, see https://en.wikipedia.org/wiki/Nvidia_Tesla