In [1]:
%load_ext Cython
In [2]:
%%cython
# distutils: extra_compile_args = -fopenmp
# distutils: extra_link_args = -fopenmp
cimport openmp
from cython.parallel cimport parallel
def get_num_threads():
cdef int num_threads
with nogil, parallel():
num_threads = openmp.omp_get_num_threads()
with gil:
return num_threads
In [3]:
get_num_threads()
Out[3]: