Question

我需要一种一致的方法来重置我的程序创建的所有线程局部变量。问题在于线程局部数据是在与使用它们的位置不同的地方创建的。

我的课程大纲如下：

struct data_t { /* ... */ };

// 1. Function that fetches the "global" thread-local data
data_t& GetData()
{
    static data_t *d = NULL;
    #pragma omp threadprivate(d); // !!!

    if (!d) { d = new data_t(); }

    return *d;
}

// 2 example function that uses the data
void user(int *elements, int num, int *output)
{
    #pragma omp parallel for shared(elements, output) if (num > 1000)
    for (int i = 0; i < num; ++i)
    {
        // computation is a heavy calculation, on memoized data
        computation(GetData()); 
    }
}

现在，我的问题是我需要一个重置数据的函数，即必须考虑创建的每个线程局部对象。

现在，我的解决方案是使用并行区域，希望使用与“parallel for”相同或更多的线程，以便每个对象都“迭代”通过：

void ClearThreadLocalData()
{
    #pragma omp parallel
    {
        // assuming data_t has a "clear()" method
        GetData().clear();
    }
}

是否有更惯用/安全的方式来实施ClearThreadLocalData()？

Answer 1

您可以为数据创建和使用全局版本号。每次需要清除现有缓存时增加它。然后修改GetData以检查版本号是否存在现有数据对象，丢弃现有数据对象并创建新数据对象（如果已过期）。（如果可以修改类，则分配的data_t对象的版本号可以存储在data_t内;如果不是，则可以存储在第二个线程局部变量中。）您最终会得到类似的东西。

static int dataVersion;

data_t& GetData()
{
    static data_t *d = NULL;
    #pragma omp threadprivate(d); // !!!

    if (d && d->myDataVersion != dataVersion) {
        delete d;
        d = nullptr;
    }
    if (!d) { 
        d = new data_t();
        d->myDataVersion = dataVersion;
    }

    return *d;
}

这并不取决于Clear中是否存在data_t方法，但如果您有一个方法，则可以通过调用Clear替换删除并重置。我使用d = nullptr来避免重复调用new data_t()。

如果要避免全局变量，全局dataVersion可以是data_t的静态成员，并且如果需要它可以是原子的，尽管GetData需要更改来处理它。

当需要重置数据时，只需更改全局版本号：

++dataVersion;

在OpenMP中重置线程局部变量

1 个答案: