View all mkl analysis

How to use the mkl.domain_get_max_threads function in mkl

To help you get started, we’ve selected a few mkl examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

IntelPython / mkl_fft / mkl_fft / _scipy_fft_backend.py View on Github

def _workers_to_num_threads(w):
    if w is None:
        return mkl.domain_get_max_threads(domain='fft')
    return int(w)

IntelPython / mkl_fft / mkl_fft / _scipy_fft_backend.py View on Github

def __exit__(self, *args):
        # restore default
        max_num_threads = mkl.domain_get_max_threads(domain='fft')
        mkl.domain_set_num_threads(max_num_threads, domain='fft')