如何使用自定义wait函数替换OpenMP的!$OMP BARRIER屏障指令

第一版代码段错误原因 #

• 参数调用不匹配：你调用wait时仅传入了1个整数参数，但wait子程序定义要求传入2个REAL64类型的omp_start、omp_end参数，参数数量、类型完全不匹配，调用时栈结构被破坏，直接触发非法内存访问导致段错误。你怀疑的INT函数使用问题不是段错误的核心诱因。
• 逻辑本身不可靠：用任务入队的耗时作为其他线程的等待时间完全不可靠，不同设备性能、运行环境负载都会影响入队耗时，要么等待时间不足导致队列未完成入队就往下执行，要么等待冗余浪费性能。

第二版修改后结果不一致原因 #

• 存在数据竞争：queue_full函数读取queue%size字段时没有加锁，而1号线程入队时会修改size字段，多线程并发读写同一个变量没有同步措施，会导致读到脏数据，queue_full的判断结果完全不可靠。
• 同步逻辑不对等：原生!$OMP BARRIER的作用是等待所有线程都执行到屏障位置才共同往下执行，你自己实现的wait仅等待队列填充满，没有保证所有线程都执行到该位置，也没有保证1号线程入队后的所有前置操作都对其他线程可见，运行结果必然和原版本不一致。
• 等待粒度过粗：每次sleep(1)的等待粒度是1秒，哪怕逻辑正确，也会导致等待延迟过高，性能远差于原生屏障。

正确的替换方案 #

如果你确实需要自定义同步逻辑替代OpenMP原生屏障，可以按如下方式实现：

1. 定义一个共享的原子标志位，保证多线程下的可见性：

logical, volatile, shared :: queue_ready = .false. ! 也可直接使用OpenMP标准原子变量

2. 1号线程完成所有入队操作后，修改标志位：

!$OMP ATOMIC WRITE
queue_ready = .true.

3. 非1号线程循环等待标志位置为真，判断用原子读：

logical :: is_ready
if (num_thread .ne. 1) then
    do
        !$OMP ATOMIC READ
        is_ready = queue_ready
        if (is_ready) exit
        call sleep(0.01) ! 用短时长休眠降低CPU占用
    end do
end if

如果要保留队列满的判断逻辑，每次读取queue%size时必须和入队操作持有同一把锁，避免数据竞争。

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