c++ - 动态可调阵列和 OpenMP

Question

我有一个函数使用 realloc 来动态调整一维数组的内存，因为数组的初始大小无法预先确定。我想通过将任务划分到多个线程来并行化此代码，每个线程将在一个线程上工作较小的一维数组，他们会根据所需的内存动态重新调整。作为该过程的一部分，每个线程还会生成一个私有(private)变量，该变量将包含小数组的最终大小。

在 Openmp 中，我想访问这些数组的私有(private)拷贝(通过主线程)，并将所有小数组放在一起，根据每个线程计算的数组大小获得最终数组。

这可能吗？？

Answer 1

这可以通过动态数组来完成，例如 std::vector。例如，假设您有一个名为 data 的数组，其值介于 0 和 1 之间以及 n 元素，您想要选择大于 0.5 的值并将它们存储在一个新的数组 vec。你可以像这样做你想做的事

double data[n];
for(int i=0; i<n; i++) data[i] = 1.0*rand()/RAND_MAX;

std::vector<double> vec;
#pragma omp parallel
{
    std::vector<double> vec_private;
    #pragma omp for nowait 
    for(int i=0; i<n; i++) {
        if(data[i]>0.5) vec_private.push_back(data[i]);
    }
    #pragma omp critcal
    vec.insert(vec.end(), vec_private.begin(), vec_private.end());
}

要在没有关键部分的情况下做到这一点需要更多的工作。它需要为每个数组保存一个大小的数组，并在单个部分中对该数组进行累加和(也称为前缀和)。一旦我们有了累积和，我们就可以用它来并行合并数组。

int *sizea;
#pragma omp parallel
{
    int nthreads = omp_get_num_threads();
    #pragma omp single
    {   
        sizea = new int [nthreads+1];
        sizea[0] = 0;
    }
    std::vector<double> vec_private;
    #pragma omp for schedule(static) nowait
    for(int i=0; i<n; i++) {
        if(data[i]>0.5) vec_private.push_back(data[i]);
    }
    sizea[omp_get_thread_num()+1] = vec_private.size();
    #pragma omp barrier
    #pragma omp single
    {
        int size = 0;
        for(int i=0; i<nthreads+1; i++) {
            size += sizea[i];
            sizea[i] = size;            
        }
        vec.resize(size);
    }
    std::copy(vec_private.begin(), vec_private.end(), vec.begin()+sizea[omp_get_thread_num()]);
}
delete[] sizea;