JUSTUS2/Software/Julia/Parallel Programming

From bwHPC Wiki
Jump to navigation Jump to search

Parallel Programming in Julia

Julia supports several paradigms of parallel programming:

  1. Implicit multi-threading by math libraries (OpenBLAS, MKL)
  2. Explicit multi-threading using Julia threads (e.g. `Threads.@threads for`)
  3. Multiple processes on one ore more nodes
  4. Execution on GPUs/CUDA using CUDA.jl

All paradigms may be used at the same time, but must be chosen carefully, to obtain the desired performance.

Implict Multi-Threading

The number of threads used by the mathematical linear algebra libraries may be configured using BLAS.set_num_threads() from the LinearAlgebra package. Alternatively you can set the environment variables OPENBLAS_NUM_THREADS or MKL_NUM_THREADS if you use MKL.

If your code is already multi-threaded, you probably want to set the number of BLAS threads to 1, in order to avoid running too many competing threads, as every Julia thread comes with its own BLAS threads.


Explicit Multi-Threading

Start Julia with option -t x where x is the number of (Julia) threads or the keyword auto, which however doesn't determine correctly the number of threads requested from SLURM with the option --cpus-per-task. Alternatively, you can set the environment variable JULIA_NUM_THREADS. See the Julia documentation for more details.