单线程下OpenMP任务归约结合目标卸载触发段错误

问题重现 #

在Clang 17.0.0 + CUDA 12.1的Ubuntu 22.04环境中，运行以下OpenMP代码时，若并行区域指定num_threads(1)会触发段错误；但将线程数设为2及以上，或移除target指令的nowait选项后，程序可正常运行：

#include <omp.h>

int main(){
    int res = 0;

    #pragma omp parallel num_threads(1)
    {
        #pragma omp single
        {
            #pragma omp taskgroup task_reduction(+:res)
            {
                #pragma omp target in_reduction(+:res) nowait
                {
                    res++;
                }
            }
        }
    }
}

编译命令：

clang++ -fopenmp -fopenmp-targets=nvptx64 --offload-arch=sm_61 -O0 main.cpp

原因分析 #

该问题大概率是Clang OpenMP运行时在单线程环境下，对异步target任务的reduction同步处理存在bug：

• 使用nowait后，主机线程不会等待target设备上的任务完成，直接推进后续逻辑；
• 单线程场景下，OpenMP运行时未正确将异步target任务纳入taskgroup的任务跟踪体系，导致taskgroup在target任务的reduction操作未完成时就提前收尾，处理reduction结果时访问无效内存，触发段错误；
• 多线程环境下，运行时的任务调度与同步机制能正常跟踪异步target任务，因此无异常；移除nowait后，主机线程会等待target任务完成，自然规避了时序问题。

解决办法 #

1. 移除nowait选项：若业务逻辑允许同步等待，直接去掉nowait，让主机线程等target设备完成reduction后再执行taskgroup收尾逻辑；
2. 强制多线程并行区域：即使实际需要单线程逻辑，也将num_threads设为2及以上，触发运行时的多线程调度逻辑，绕开单线程下的同步bug；
3. 升级Clang版本：该问题可能是Clang 17的特定版本bug，尝试升级到Clang 18或更高版本，查看官方是否已修复该单线程offload reduction同步问题。

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