OpenMP中可重用的私有动态分配数组

我使用OpenMP和MPI来并行化c中的一些矩阵运算.在矩阵上运行的一些函数是用Fortran编写的. Fortran函数需要传递一个缓冲区数组,该数组仅在函数内部使用.目前我在每个并行部分分配缓冲区,类似于下面的代码.

int i = 0;
int n = 1024; // Actually this is read from command line
double **a = createNbyNMat(n);
#pragma omp parallel
{
    double *buf;
    buf = malloc(sizeof(doublE)*n);
#pragma omp for
    for (i=0; i < n; i++)
    {
        fortranFunc1_(a[i],&n,buf);
    }
    free(z);
}

// serial code and moving data around in the matrix a using MPI

#pragma omp parallel
{
    double *buf;
    buf = malloc(sizeof(doublE)*n);
#pragma omp for
    for (i=0; i < n; i++)
    {
        fortranFunc2_(a[i],buf);
    }
    free(z);
}

// and repeat a few more times.

我知道使用类似于下面的代码的方法可以避免重新分配缓冲区,但我很好奇是否有更简单的方法或OpenMP中的一些内置功能来处理这个问题.无论是否在我们正在编译的系统上存在OpenMP,能够在没有大量编译器指令的情况下编译代码将是很好的.

double **buf;
buf = malloc(sizeof(double*) * num_openmp_threads);
int i = 0;
for (i = 0; i < num_openmp_threads; ++i)
{
    buf[i] = malloc(sizeof(doublE) * n);
}

// skip ahead

#pragma omp for
for (i=0; i < n; i++)
{
    fortranFunc1_(a[i],buf[current_thread_num]);
}

void func(...)
{
   static double *buf;
   #pragma omp threadprivate(buf)

   #pragma omp parallel num_threads(nth)
   {
       buf = malloc(n * sizeof(doublE));
       ...
   }

   #pragma omp parallel num_threads(nth)
   {
       // Access buf here - it is still allocated
   }

   #pragma omp parallel num_threads(nth)
   {
       // Free the memory in the last parallel region
       free(buf);
   }
}