如何在循环内将大对象复制到omp任务?

时间:2019-07-15 11:26:27

标签: c++ openmp

我正在使用while循环来启动几个omp任务。每个任务都需要复制一个相当大的对象(作为第一个私有对象)。由于我的设置,大对象(在此示例中为矢量)将被天真地复制两次:

struct bigStruct {
    bool next() {
        /* do something with m_bigVector */
    }

    std::vector<int> m_bigVector;
    /* other (big) data members */
};

bigStruct s;

#pragma omp parallel
{
    #pragma omp single
    while (s.next()) {
        auto obj = s.m_bigVector; //copy the first time

        #pragma omp task firstprivate(obj) //copy the second time
        {
            /* do something with obj */
        }
    }
} //end parallel

gcc优化(-O3)似乎没有以任何方式优化两个复制步骤。 一个(不太优雅的)解决方案是使用显式new/delete

#pragma omp parallel
{
    #pragma omp single
    while (s.next()) {
        auto obj_ptr = new std::vector<int>(s.m_bigVector); //copy once

        #pragma omp task firstprivate(obj_ptr) //copy only the pointer
        {
            /* do something with obj */


            delete obj_ptr; 
        }
    }
} //end parallel

是否有更多modern/elegant方法可以解决此问题?也许是一种告诉任务移动对象而不是复制对象的方法? 请注意,我不想复制整个bigStruct,因为它包含其他大数据成员。

1 个答案:

答案 0 :(得分:1)

好消息!

  

firstprivate变量不能具有引用类型。

自OpenMP 4.5(2015)起已过时。目前没有这样的限制。有一个要求:

  

如果工作共享构造的firstprivate子句中的列表项具有引用类型,则它必须绑定到团队所有线程的同一对象。

但这并不适用-task构造不是工作共享构造,并且无论如何也不会被多个线程遇到。

要完全了解标准的要求,请执行以下操作:

(关于列表项私有化)

  

如果列表项的类型是对类型T的引用,则出于本条款的所有目的,该类型将被视为T

     

为该结构分配了一个具有自动存储持续时间的相同类型的新列表项。   这些列表项的存储和生存期一直持续到创建它们的块退出为止。

     

对于每个类类型的变量:

     

•如果firstprivate子句不在target构造上,则调用复制构造函数执行初始化

因此您可以放心地做:

auto& obj = s.m_bigVector;
#pragma omp task firstprivate(obj) // call copy ctor once

不幸的是你不能

  • 使用const auto&是因为obj的类型将是常量,因为仅删除了引用。
  • 将obj移动到firstprivate声明中。那样很好,但是仅适用于只有单个线程实际遇到数据共享子句的任务。