Cython中如何等待prange启动的所有线程全部执行完毕？

Cython的prange本身自带隐式等待逻辑，不需要额外调用join类方法，当prange循环执行结束时，所有由prange启动的工作线程的任务已经全部执行完成，主线程会自动在循环结束位置等待所有任务完成后再向下执行后续代码。 #

核心逻辑说明 #

• prange底层基于OpenMP实现并行逻辑，OpenMP的并行循环默认在循环末尾设置了隐式屏障（implicit barrier），会阻塞主线程直到所有工作线程完成分配的迭代任务，才会释放屏障继续向下执行。
• 你示例代码里的cython.parallel.join_all_prange_threads()是完全不需要的冗余调用，直接在prange循环结束后编写依赖执行结果的业务代码即可。

注意：只有当你显式给prange传入nowait=True参数关闭默认隐式屏障时，才会出现循环结束后任务未完成的情况，常规用法下不需要做额外等待处理。

修正后的代码示例 #

from cython.parallel import prange

cdef int i
cdef int n = len(data)
for i in prange(n, nogil=True, num_threads=10):
    # 每个线程独立处理data[i]的逻辑
    pass    

# 到这里所有prange启动的线程任务已全部完成，无需额外等待
do_something_with_results_by_prange_threads()

特殊场景处理 #

如果你确实碰到prange循环结束后还有未完成的任务，说明你是在prange的迭代逻辑中手动启动了不受prange管理的游离线程，这种场景可以用以下方式实现等待：

• 定义一个线程安全的计数器，初始值为你启动的游离线程总数
• 每个游离线程执行完成后原子性将计数器减1
• 主线程在prange循环后循环判断计数器是否为0，为0时再向下执行

内容的提问来源于stack exchange，提问作者Dan D