Source code for mtalg.tools.__set_threads

import mtalg




def set_num_threads(num_threads: int):
    """Set number of threads for subsequent multithreaded RNGs and algebra functions.

    Args:
        num_threads: Number of threads

    Examples:
        Check the available number of threads.

        >>> from multiprocessing import cpu_count
        >>> cpu_count()
        12

        Check the number of threads used by mtalg.

        >>> mtalg.get_num_threads()
        12

        Change the number of threads used by mtalg.

        >>> mtalg.set_num_threads(6)
        >>> mtalg.get_num_threads()
        6
    """
    if not isinstance(num_threads, int):
        raise ValueError(f'Number of threads must be an integer, found: {num_threads}')
    if not num_threads > 0:
        raise ValueError(f'Number of threads must be > 0, found: {num_threads}')

    mtalg.core.threads._global_num_threads = num_threads
    mtalg.random._RNG = mtalg.random.MultithreadedRNG(num_threads=num_threads)