2

我正在使用 mex 文件测试基本的 openmp 并行代码。问题是它似乎只运行一个线程,尽管我的指示是用 2 个线程运行它。这是代码:

#include "mex.h"
#include "omp.h"

#include <iostream>


void mexFunction(int nlhs, mxArray *plhs[],int nrhs,const mxArray *prhs[])
{
    using namespace std;
    #define x_out plhs[0]
    #define x_in prhs[0]

    double *x;
    double y;
    x_out=mxCreateDoubleMatrix(1,1,mxREAL);
    x=mxGetPr(x_out);
    y=mxGetScalar(x_in);

    x[0]=y;    
    omp_set_num_threads(2);
    int Nthreads=omp_get_num_threads();
    cout<<Nthreads<<"\n";
    #pragma omp parallel
    {
        int ithread=omp_get_thread_num();

        #pragma omp for
                for (int i=0;i<10;i++)
                    cout<<"Hello! " <<i<<"\n";
    }
    return;
}

我使用以下编译行 -

mex -v paralletestmex.cpp CC=g++ CFLAGS="\$CFLAGS -fopenmp" LDFLAGS="\$LDFLAGS -fopenmp"

并且在详细描述中显示了 fopenmp 标志,所以我猜它正在将其编译为并行。

我得到的输出是 -

1
Hello! 0
Hello! 1
Hello! 2
Hello! 3
Hello! 4
Hello! 5
Hello! 6
Hello! 7
Hello! 8
Hello! 9

表明由于某种原因,只创建了 1 个线程。这是针对我在更复杂代码上面临的问题的简单测试。当我在不使用 mex 的情况下将其作为纯 c++ 文件运行时,相同的代码似乎可以正常工作。

任何帮助表示赞赏。谢谢!悉达多

4

2 回答 2

3

这是一个极其常见的错误:omp_get_num_threads()返回当前团队中的线程数。当在区域外调用时parallel,它总是返回 1,因为根据定义,OpenMP 程序在并行区域外仅使用单个线程(主线程)执行。

的补充调用omp_set_num_threads()omp_get_max_threads().

另请注意,omp_set_num_threads()在编写模块和库函数时,调用是一种非常糟糕的编程习惯。原因是它固定了所有并行区域的线程数,因此可能会影响其他代码。一个更好的方法是使用该num_threads子句:

#pragma omp parallel num_threads(2)
{
   // ...
}
于 2013-10-29T08:42:43.300 回答
2

好的,我做了一些研究,结果发现 mexopts.sh 文件中的 CXXOPTIMFLAGS 也需要更改。所以我在编译行中添加了:

CXXOPTIMFLAGS="\$CXXOPTIMFLAGS -fopenmp" 

这似乎可以完成这项工作。

感谢你的帮助!

于 2013-10-29T20:20:14.947 回答