libgomp/testsuite/libgomp.c/teams-2.c

   1 #include <omp.h>
   2 #include <stdlib.h>
   3
   4 __attribute__((noinline))
   5 void
   6 foo (int x, int y, int z, int *a, int *b)
   7 {
   8   if (x == 0)
   9     {
  10       int i, j;
  11       for (i = 0; i < 64; i++)
  12         #pragma omp parallel for shared (a, b)
  13         for (j = 0; j < 32; j++)
  14           foo (3, i, j, a, b);
  15     }
  16   else if (x == 1)
  17     {
  18       int i, j;
  19       #pragma omp distribute dist_schedule (static, 1)
  20       for (i = 0; i < 64; i++)
  21         #pragma omp parallel for shared (a, b)
  22         for (j = 0; j < 32; j++)
  23           foo (3, i, j, a, b);
  24     }
  25   else if (x == 2)
  26     {
  27       int j;
  28       #pragma omp parallel for shared (a, b)
  29       for (j = 0; j < 32; j++)
  30         foo (3, y, j, a, b);
  31     }
  32   else
  33     {
  34       #pragma omp atomic
  35       b[y] += z;
  36       #pragma omp atomic
  37       *a += 1;
  38     }
  39 }
  40
  41 __attribute__((noinline))
  42 int
  43 bar (int x, int y, int z)
  44 {
  45   int a, b[64], i;
  46   a = 8;
  47   for (i = 0; i < 64; i++)
  48     b[i] = i;
  49   foo (x, y, z, &a, b);
  50   if (x == 0)
  51     {
  52       if (a != 8 + 64 * 32)
  53         return 1;
  54       for (i = 0; i < 64; i++)
  55         if (b[i] != i + 31 * 32 / 2)
  56           return 1;
  57     }
  58   else if (x == 1)
  59     {
  60       int c = omp_get_num_teams ();
  61       int d = omp_get_team_num ();
  62       int e = d;
  63       int f = 0;
  64       for (i = 0; i < 64; i++)
  65         if (i == e)
  66           {
  67             if (b[i] != i + 31 * 32 / 2)
  68               return 1;
  69             f++;
  70             e = e + c;
  71           }
  72         else if (b[i] != i)
  73           return 1;
  74       if (a < 8 || a > 8 + f * 32)
  75         return 1;
  76     }
  77   else if (x == 2)
  78     {
  79       if (a != 8 + 32)
  80         return 1;
  81       for (i = 0; i < 64; i++)
  82         if (b[i] != i + (i == y ? 31 * 32 / 2 : 0))
  83           return 1;
  84     }
  85   else if (x == 3)
  86     {
  87       if (a != 8 + 1)
  88         return 1;
  89       for (i = 0; i < 64; i++)
  90         if (b[i] != i + (i == y ? z : 0))
  91           return 1;
  92     }
  93   return 0;
  94 }
  95
  96 int
  97 main ()
  98 {
  99   int i, j, err = 0;
 100   #pragma omp teams reduction(+:err)
 101   err += bar (0, 0, 0);
 102   if (err)
 103     abort ();
 104   #pragma omp teams reduction(+:err)
 105   err += bar (1, 0, 0);
 106   if (err)
 107     abort ();
 108   #pragma omp teams reduction(+:err)
 109   #pragma omp distribute
 110   for (i = 0; i < 64; i++)
 111     err += bar (2, i, 0);
 112   if (err)
 113     abort ();
 114   #pragma omp teams reduction(+:err)
 115   #pragma omp distribute
 116   for (i = 0; i < 64; i++)
 117   #pragma omp parallel for reduction(+:err)
 118     for (j = 0; j < 32; j++)
 119       err += bar (3, i, j);
 120   if (err)
 121     abort ();
 122   return 0;
 123 }