liger_iris_sim.utils.parallelization_utils

Functions

numba_thread_scope(num_threads)

Context manager to temporarily set the number of threads used by numba, and reset when finished.

numba_thread_scope(num_threads: int)[source]

Context manager to temporarily set the number of threads used by numba, and reset when finished.

Parameters:

num_threads – The number of threads to set.

Examples

with numba_thread_scope(num_threads):

# Numba calls here