python - cython 中的关键部分

Question

在cython中使用prange时如何处理cython中的临界区？

是否允许#pragma omp critical？

for e in prange(num,nogil=True,num_threads=4):
.
.
.
  #pragma omp critical
  cs
.
.
.
end_for

Answer 1

要处理临界区，您可以使用 openmp 锁。为此，使用

导入它

cimport openmp

然后初始化

cdef openmp.omp_lock_t lock
openmp.omp_init_lock(&lock)

然后这样使用

with nogil, cython.boundscheck(False), cython.wraparound(False):
    for i in parallel.prange(n, schedule = 'static', 
                             num_threads = num_threads):
        .
        .
        .
        openmp.omp_set_lock(&lock)
        cs
        openmp.omp_unset_lock(&lock)
        .
        .
        .