OpenMP set_num_threads() is not working

OpenMP set_num_threads() is not working

本文关键字:is not working threads set num OpenMP      更新时间:2023-10-16

我正在C++中使用OpenMP编写一个并行程序。

我想使用 omp_set_num_threads() 控制程序中的线程数,但它不起作用。

#include <iostream>
#include <omp.h>
#include "mpi.h"
using namespace std;
int myrank;
int groupsize;
double sum;
double t1,t2;
int n = 10000000;
int main(int argc, char *argv[])
{
    MPI_Init( &argc, &argv);
    MPI_Comm_rank( MPI_COMM_WORLD, &myrank );
    MPI_Comm_size(MPI_COMM_WORLD,&groupsize);
    omp_set_num_threads(4);
    sum = 0;
    #pragma omp for  reduction(+:sum)
    for (int i = 0; i < n; i++)
        sum+= i/(n/10);
    cout<<"sum="<<sum<<endl;
    cout<<"threads="<<omp_get_num_threads()<<endl;
    MPI_Finalize();
    return 0;
}

程序输出:

sum = 4.5e+007
threads=1

如何控制线程数?

除了在

并行区域之外调用omp_get_num_threads()之外,调用omp_set_num_threads()仍然不能保证 OpenMP 运行时将完全使用指定数量的线程。 omp_set_num_threads() 用于覆盖环境变量的值OMP_NUM_THREADS它们都控制 OpenMP 将为所有并行区域(在 OMP_NUM_THREADS 的情况下)或任何后续并行区域(在调用 omp_set_num_threads() 之后)生成的线程组大小的上限。有一种称为动态团队的东西,如果运行时系统认为它更合适,它仍然可以选择较少数量的线程。您可以通过调用omp_set_dynamic(0)或将环境变量OMP_DYNAMIC设置为 false 来禁用动态团队。

要强制实施给定数量的线程,您应该禁用动态团队并使用任一omp_set_num_threads()指定所需的线程数:

omp_set_dynamic(0);     // Explicitly disable dynamic teams
omp_set_num_threads(4); // Use 4 threads for all consecutive parallel regions
#pragma omp parallel ...
{
    ... 4 threads used here ...
}

或使用 num_threads OpenMP 子句:

omp_set_dynamic(0);     // Explicitly disable dynamic teams
// Spawn 4 threads for this parallel region only
#pragma omp parallel ... num_threads(4)
{
    ... 4 threads used here ...
}

omp_get_num_threads()函数返回当前在执行调用它的并行区域的组中的线程数。您在并行区域之外调用它,这就是它返回 1 的原因。

根据 GCC 的omp_get_num_threads手册:

在程序的顺序部分中,omp_get_num_threads返回 1

所以这个:

cout<<"sum="<<sum<<endl;
cout<<"threads="<<omp_get_num_threads()<<endl;

应更改为以下内容:

#pragma omp parallel
{
    cout<<"sum="<<sum<<endl;
    cout<<"threads="<<omp_get_num_threads()<<endl;
}

我使用的代码也遵循了 Hristo 的建议,即禁用动态团队。

我遇到了同样的问题.解决方案如下

右键单击"源程序">"属性">"配置属性"> C/C++> 语言> 现在将"打开 MP 支持标志"更改为"是..."。

你会得到想要的结果。

相关文章: