web-dev-qa-db-ja.com

breakステートメントを使用した並列OpenMPループ

OpenMPループのbreakステートメントを使用できないことは知っていますが、並列処理のメリットを活かしながら、回避策があるかどうか疑問に思っていました。基本的に私は 'for'ループを持っています。これは、特定の条件を満たす1つの要素を探す大きなベクトルの要素をループします。ただし、条件を満たす要素は1つだけなので、それが見つかったらループから抜けることができます。よろしくお願いします。

for(int i = 0; i <= 100000; ++i)
  {
    if(element[i] ...)
     {
          ....
          break;
      }
   }
22
Jeanno

あなたはwhileループを使用して、openmp forループが行うことを手動で行うことを試みることができます:

const int N = 100000;
std::atomic<bool> go(true);
uint give = 0;

#pragma omp parallel
{
    uint i, stop;

    #pragma omp critical
    {
        i = give;
        give += N/omp_get_num_threads();
        stop = give;

        if(omp_get_thread_num() == omp_get_num_threads()-1)
            stop = N;
    } 


    while(i < stop && go)
    {
        ...
        if(element[i]...)
        {
            go = false;
        }
        i++;
    }
}

この方法では、サイクルごとに "go"をテストする必要がありますが、それほど重要ではありません。さらに重要なのは、これが「静的な」omp forループに対応することです。これは、すべての反復に同じ量の時間がかかると予想できる場合にのみ役立ちます。そうしないと、3つのスレッドがすでに完了している可能性があります。

8
Haatschii

このスニペットを参照してください:

volatile bool flag=false;

#pragma omp parallel for shared(flag)
for(int i=0; i<=100000; ++i)
{    
    if(flag) continue;
    if(element[i] ...)
    {
          ...
          flag=true;
    }
}

この状況は、pthreadに適しています。

24
yyfn

私はおそらくそうするでしょう(yyfnから少しコピーした)

volatile bool flag=false;

for(int j=0; j<=100 && !flag; ++j) {
  int base = 1000*j;
  #pragma omp parallel for shared(flag)
  for(int i = 0; i <= 1000; ++i)
  {

    if(flag) continue;
    if(element[i+base] ...)
     {
          ....
          flag=true;
      }
   }
}
2
enobayram

受け入れられた回答の簡単なバージョンを以下に示します。

int ielement = -1;
#pragma omp parallel
{
    int i = omp_get_thread_num()*n/omp_get_num_threads();
    int stop = (omp_get_thread_num()+1)*n/omp_get_num_threads();        
    for(;i <stop && ielement<0; ++i){
        if(element[i]) {
            ielement = i;
        }
    }
}
0
Z boson
bool foundCondition = false;
#pragma omp parallel for
for(int i = 0; i <= 100000; i++)
{
    // We can't break out of a parallel for loop, so this is the next best thing.
    if (foundCondition == false && satisfiesComplicatedCondition(element[i]))
    {
        // This is definitely needed if more than one element could satisfy the
        // condition and you are looking for the first one.  Probably still a
        // good idea even if there can only be one.
        #pragma omp critical
        {
            // do something, store element[i], or whatever you need to do here
                ....

            foundCondition = true;
        }
    }
}
0
Peter Smartt