OpenMP中sections内嵌套parallel for无法并行的实现方案咨询

1. 嵌套并行默认禁用：OpenMP默认关闭嵌套并行功能，当你在parallel sections的section内部再使用parallel for时，内层的并行区域不会创建新线程，只会复用当前section的线程串行执行循环。
2. 线程号的作用域：omp_get_thread_num()返回的是当前所在并行区域内的线程编号，而非全局线程编号。内层并行区域因未创建新线程，只有1个线程，所以始终输出0。

方式一：启用嵌套并行 #

通过代码或环境变量开启嵌套并行，让内层parallel for可以创建新线程：

代码修改示例 #

#include <iostream>
#include <vector>
#include <omp.h>

int main() {
    // 启用嵌套并行
    omp_set_nested(1);

#pragma omp parallel sections num_threads(10)
    {
#pragma omp section
        {
            std::vector<int> a(5);
            // 内层并行循环指定线程数（可选，不指定则用默认值）
#pragma omp parallel for num_threads(4)
            for (int i = 0; i < a.size(); i++) {
                // 这里的线程号是内层并行区域的编号（0-3）
                std::cout << "Section 1, inner thread: " << omp_get_thread_num() << std::endl;
                a[i] = i;
            }
        }
#pragma omp section
        {
            std::vector<int> b(5);
#pragma omp parallel for num_threads(4)
            for (int i = 0; i < b.size(); i++) {
                std::cout << "Section 2, inner thread: " << omp_get_thread_num() << std::endl;
                b[i] = i;
            }
        }
    }
    return 0;
}

环境变量方式 #

无需修改代码，运行程序前设置环境变量：

export OMP_NESTED=true
./your_program

方式二：复用外层线程池（避免嵌套） #

如果不想启用嵌套并行，可以在外层创建一个大的并行区域，然后在section内部用omp for（而非parallel for）将循环任务分配到外层的线程池中：

#include <iostream>
#include <vector>
#include <omp.h>

int main() {
#pragma omp parallel num_threads(10)
    {
#pragma omp sections
        {
#pragma omp section
            {
                std::vector<int> a(5);
                // 用omp for分配循环到外层并行区域的线程
#pragma omp for
                for (int i = 0; i < a.size(); i++) {
                    // 这里的线程号是外层并行区域的编号（0-9）
                    std::cout << "Section 1, outer thread: " << omp_get_thread_num() << std::endl;
                    a[i] = i;
                }
            }
#pragma omp section
            {
                std::vector<int> b(5);
#pragma omp for
                for (int i = 0; i < b.size(); i++) {
                    std::cout << "Section 2, outer thread: " << omp_get_thread_num() << std::endl;
                    b[i] = i;
                }
            }
        }
    }
    return 0;
}

这种方式下，两个section的循环会共享外层的10个线程池，既实现了section之间的并行，也实现了每个循环内部的并行，同时避免了嵌套并行的开销。

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