如何使用 Mexcallmatlab 调用用户定义的函数？

Question

我正在尝试在 mex 文件中使用 OpenMP 并行化我的一段 Matlab 代码。 Matlab 代码中我要并行化的部分是：

对于 i = 1 : n

    D(:, i) = CALC(A, B(:,i), C(i));

end

我写这个是为了并行化它：

 void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
size_t r,n,i,G;
double *A, *B, *C, *D;
int nthreads;
nthreads = 4;
A = mxGetPr(prhs[0]); /* first input matrix */
B = mxGetPr(prhs[1]); /* second input matrix */
C = mxGetPr(prhs[2]);/* third input matrix */
/* dimensions of input matrices */
r = mxGetN(prhs[0]);
n = mxGetN(prhs[1]);
plhs[0] = mxCreateDoubleMatrix(r,n, mxREAL);
D = mxGetPr(plhs[0]);
G=n/nthreads;
omp_set_num_threads(nthreads);
#pragma omp parallel for schedule (dynamic, G)
  {
   for i = 1 : n
        D(:, i) = CALC(A, B(:,i), C(i));
  }
 }

CALC 是我编写的一个 Matlab 函数。我的挑战是如何使用 Mexcallmatlab 将 CALC 函数调用到 mex 文件中，以便它可以在我的 mex 文件中并行执行它，并将 D 的每一列的元素（即 D(:, i) 返回到我的Matlab代码。

很抱歉这个冗长的问题。非常感谢我能得到的任何帮助。

Answer 1

您需要使用多个 MATLAB 进程才能并行运行多个调用。最简单的方法是使用并行计算工具箱并使用 parfor 而不是 for 循环。