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  }