Multi-Threading

    By default, Julia starts up with a single thread of execution. This can be verified by using the command Threads.nthreads():

    The number of execution threads is controlled either by using the /--threads command line argument or by using the environment variable. When both are specified, then -t/--threads takes precedence.

    The number of threads can either be specified as an integer (--threads=4) or as auto (--threads=auto), where auto tries to infer a useful default number of threads to use (see Command-line Options for more details).

    Julia 1.5

    The -t/--threads command line argument requires at least Julia 1.5. In older versions you must use the environment variable instead.

    Julia 1.7

    Using auto as value of the environment variable JULIA_NUM_THREADS requires at least Julia 1.7. In older versions, this value is ignored.

    Lets start Julia with 4 threads:

    1. $ julia --threads 4

    Let’s verify there are 4 threads at our disposal.

    1. julia> Threads.nthreads()
    2. 4

    But we are currently on the master thread. To check, we use the function 

    1. julia> Threads.threadid()
    2. 1

    Note

    If you prefer to use the environment variable you can set it as follows in Bash (Linux/macOS):

    1. export JULIA_NUM_THREADS=4

    C shell on Linux/macOS, CMD on Windows:

    1. set JULIA_NUM_THREADS=4

    Powershell on Windows:

    Note that this must be done before starting Julia.

    Note

    When a program’s threads are busy with many tasks to run, tasks may experience delays which may negatively affect the responsiveness and interactivity of the program. To address this, you can specify that a task is interactive when you Threads.@spawn it:

    1. using Base.Threads
    2. @spawn :interactive f()

    Interactive tasks should avoid performing high latency operations, and if they are long duration tasks, should yield frequently.

    Julia may be started with one or more threads reserved to run interactive tasks:

    1. $ julia --threads 3,1

    The environment variable JULIA_NUM_THREADS can also be used similarly:

    1. export JULIA_NUM_THREADS=3,1

    This starts Julia with 3 threads in the :default threadpool and 1 thread in the :interactive threadpool:

    1. julia> using Base.Threads
    3. julia> nthreads()
    4. 4
    6. julia> nthreadpools()
    7. 2
    9. julia> threadpool()
    10. :default
    12. julia> nthreads(:interactive)
    13. 1

    Either or both numbers can be replaced with the word auto, which causes Julia to choose a reasonable default.

    Communication and synchronization

    Although Julia’s threads can communicate through shared memory, it is notoriously difficult to write correct and data-race free multi-threaded code. Julia’s s are thread-safe and may be used to communicate safely.

    You are entirely responsible for ensuring that your program is data-race free, and nothing promised here can be assumed if you do not observe that requirement. The observed results may be highly unintuitive.

    The best way to ensure this is to acquire a lock around any access to data that can be observed from multiple threads. For example, in most cases you should use the following code pattern:

    1. julia> lock(lk) do
    2.            use(a)
    3.        end
    5. julia> begin
    6.            lock(lk)
    7.            try
    9.                unlock(lk)
    10.            end
    11.        end

    where lk is a lock (e.g. ReentrantLock()) and a data.

    Additionally, Julia is not memory safe in the presence of a data race. Be very careful about reading any data if another thread might write to it! Instead, always use the lock pattern above when changing data (such as assigning to a global or closure variable) accessed by other threads.

    Let’s work a simple example using our native threads. Let us create an array of zeros:

    1. julia> a = zeros(10)
    2. 10-element Vector{Float64}:
    3.  0.0
    4.  0.0
    5.  0.0
    6.  0.0
    7.  0.0
    8.  0.0
    9.  0.0
    10.  0.0
    11.  0.0
    12.  0.0

    Let us operate on this array simultaneously using 4 threads. We’ll have each thread write its thread ID into each location.

    Julia supports parallel loops using the Threads.@threads macro. This macro is affixed in front of a for loop to indicate to Julia that the loop is a multi-threaded region:

    1. julia> Threads.@threads for i = 1:10
    2.            a[i] = Threads.threadid()
    3.        end

    The iteration space is split among the threads, after which each thread writes its thread ID to its assigned locations:

    1. julia> a
    2. 10-element Vector{Float64}:
    3.  1.0
    4.  1.0
    5.  1.0
    6.  2.0
    7.  2.0
    8.  2.0
    9.  3.0
    10.  3.0
    11.  4.0
    12.  4.0

    Note that does not have an optional reduction parameter like @distributed.

    Atomic Operations

    1. julia> i = Threads.Atomic{Int}(0);
    3. julia> ids = zeros(4);
    5. julia> old_is = zeros(4);
    7. julia> Threads.@threads for id in 1:4
    8.            old_is[id] = Threads.atomic_add!(i, id)
    9.            ids[id] = id
    10.        end
    12. julia> old_is
    13. 4-element Vector{Float64}:
    14.  1.0
    15.  7.0
    16.  3.0
    18. julia> i[]
    19.  10
    21. julia> ids
    22. 4-element Vector{Float64}:
    23.  1.0
    24.  2.0
    25.  3.0
    26.  4.0

    Had we tried to do the addition without the atomic tag, we might have gotten the wrong answer due to a race condition. An example of what would happen if we didn’t avoid the race:

    1. julia> using Base.Threads
    3. julia> Threads.nthreads()
    4. 4
    6. julia> acc = Ref(0)
    7. Base.RefValue{Int64}(0)
    9. julia> @threads for i in 1:1000
    10.           acc[] += 1
    11.        end
    13. julia> acc[]
    14. 926
    16. julia> acc = Atomic{Int64}(0)
    17. Atomic{Int64}(0)
    19. julia> @threads for i in 1:1000
    20.           atomic_add!(acc, 1)
    21.        end
    23. julia> acc[]
    24. 1000

    Per-field atomics

    We can also use atomics on a more granular level using the , @atomicswap, and macros.

    Specific details of the memory model and other details of the design are written in the Julia Atomics Manifesto, which will later be published formally.

    Any field in a struct declaration can be decorated with @atomic, and then any write must be marked with @atomic also, and must use one of the defined atomic orderings (:monotonic, :acquire, :release, :acquire_release, or :sequentially_consistent). Any read of an atomic field can also be annotated with an atomic ordering constraint, or will be done with monotonic (relaxed) ordering if unspecified.

    Julia 1.7

    Per-field atomics requires at least Julia 1.7.

    When using multi-threading we have to be careful when using functions that are not pure as we might get a wrong answer. For instance functions that have a by convention modify their arguments and thus are not pure.

    External libraries, such as those called via ccall, pose a problem for Julia’s task-based I/O mechanism. If a C library performs a blocking operation, that prevents the Julia scheduler from executing any other tasks until the call returns. (Exceptions are calls into custom C code that call back into Julia, which may then yield, or C code that calls jl_yield(), the C equivalent of .)

    The @threadcall macro provides a way to avoid stalling execution in such a scenario. It schedules a C function for execution in a separate thread. A threadpool with a default size of 4 is used for this. The size of the threadpool is controlled via environment variable UV_THREADPOOL_SIZE. While waiting for a free thread, and during function execution once a thread is available, the requesting task (on the main Julia event loop) yields to other tasks. Note that @threadcall does not return until the execution is complete. From a user point of view, it is therefore a blocking call like other Julia APIs.

    It is very important that the called function does not call back into Julia, as it will segfault.

    @threadcall may be removed/changed in future versions of Julia.

    Caveats

    At this time, most operations in the Julia runtime and standard libraries can be used in a thread-safe manner, if the user code is data-race free. However, in some areas work on stabilizing thread support is ongoing. Multi-threaded programming has many inherent difficulties, and if a program using threads exhibits unusual or undesirable behavior (e.g. crashes or mysterious results), thread interactions should typically be suspected first.

    There are a few specific limitations and warnings to be aware of when using threads in Julia:

    • Base collection types require manual locking if used simultaneously by multiple threads where at least one thread modifies the collection (common examples include push! on arrays, or inserting items into a Dict).
    • The schedule used by @spawn is nondeterministic and should not be relied on.
    • Compute-bound, non-memory-allocating tasks can prevent garbage collection from running in other threads that are allocating memory. In these cases it may be necessary to insert a manual call to GC.safepoint() to allow GC to run. This limitation will be removed in the future.
    • Avoid running top-level operations, e.g. include, or eval of type, method, and module definitions in parallel.
    • Be aware that finalizers registered by a library may break if threads are enabled. This may require some transitional work across the ecosystem before threading can be widely adopted with confidence. See the next section for further details.

    Because finalizers can interrupt any code, they must be very careful in how they interact with any global state. Unfortunately, the main reason that finalizers are used is to update global state (a pure function is generally rather pointless as a finalizer). This leads us to a bit of a conundrum. There are a few approaches to dealing with this problem:

    1. When single-threaded, code could call the internal jl_gc_enable_finalizers C function to prevent finalizers from being scheduled inside a critical region. Internally, this is used inside some functions (such as our C locks) to prevent recursion when doing certain operations (incremental package loading, codegen, etc.). The combination of a lock and this flag can be used to make finalizers safe.