04_target_teams_distribute.cpp

#include <stdio.h>
#include <omp.h>
#define N 20
int main( int argv, char** argc ) {
  int x[N];
  int y[N];
  int errors = 0;
  // initialize
  for(int i=0; i<N; i++ ) {
      y[i] = 0;
      x[i] = i;
    }

#pragma omp target teams map(from:y[:N]) map(to:x[:N]) num_teams(2)
#pragma omp distribute
  for(int i=0; i<N; i++ )
    {
      y[i] = x[i];
      int team = omp_get_team_num();
      int nteams = omp_get_num_teams();
      int tid = omp_get_thread_num(); // always 0
      int nthreads = omp_get_num_threads();  // always 1

      printf( "Iteration %d given to thread %d from team %d.\n",
	      i,tid, team );

    }

  // error check
  for(int i=0; i<N; i++ )
    {
      if( y[i] != x[i] ) errors++;
    }

  if( errors != 0 )
    printf("Failed!\n");
  else
    printf("Success!\n");

  return 0;
}