如何在 OpenMP 中使用锁？

Question

我有两段 C++ 代码在 2 个不同的内核上运行。他们都写入同一个文件。

如何使用OpenMP并确保不崩溃？

Answer 1

您需要 OMP_SET_LOCK/OMP_UNSET_LOCK 函数：
https://hpc.llnl.gov/tuts/openMP/#OMP_SET_LOCK

基本上：

omp_lock_t writelock;

omp_init_lock(&writelock);

#pragma omp parallel for
for ( i = 0; i < x; i++ )
{
    // some stuff
   omp_set_lock(&writelock);
    // one thread at a time stuff
    omp_unset_lock(&writelock);
    // some stuff
}

omp_destroy_lock(&writelock);

尽管具体的 API 调用不同，但大多数锁定例程（例如 pthreads 信号量和 sysv 信号量）都适用于此类逻辑。

Answer 2

为了后来者的利益，使用关键是另一种选择。您甚至可以创建命名的关键部分。

例如：

#include <omp.h>

void myParallelFunction()
{
    #pragma omp parallel for
    for(int i=0;i<1000;++i)
    {

        // some expensive work 

        #pragma omp critical LogUpdate
        {
            // critical section where you update file        
        }

        // other work

        #pragma omp critical LogUpdate
        {
            // critical section where you update file      
        }
    }
} 

编辑：Victor Eijkhout 发起的评论中有一个很棒的线索。总结和释义：简而言之，关键性锁定了一个代码段。在更复杂的示例中，您只想锁定特定的数据项，这可能有点过分了。在这两种方法之间做出选择之前，了解这一点很重要。

Answer 3

#pragma omp critical
{
    // write to file here
}