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