aboutsummaryrefslogtreecommitdiff
path: root/libgomp/testsuite/libgomp.c-c++-common/teams-2.c
blob: 316bcfe58486874d948fc9686e6708ade48d5124 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include <omp.h>
#include <stdlib.h>

int
foo ()
{
  return 934;
}

int
main ()
{
  int a[934] = {};
  int k, e;
  #pragma omp target map(a)
  #pragma omp teams num_teams (foo ())
  {
    int i = omp_get_team_num ();
    if (omp_get_num_teams () != 934
	|| (unsigned) i >= 934U
	|| a[i] != 0)
      abort ();
    ++a[i];
  }
  #pragma omp target map(a)
  #pragma omp teams num_teams (foo () - 50 : foo ())
  {
    int i = omp_get_team_num ();
    int j = omp_get_num_teams ();
    if (j < 884
	|| j > 934
	|| (unsigned) i >= (unsigned) j
	|| a[i] != 1)
      abort ();
    ++a[i];
  }
  #pragma omp target teams map(a) num_teams (foo () / 2)
  {
    int i = omp_get_team_num ();
    if (omp_get_num_teams () != 467
	|| (unsigned) i >= 467U
	|| a[i] != 2)
      abort ();
    ++a[i];
  }
  #pragma omp target teams map(a) num_teams (foo () / 2 - 50 : foo () / 2)
  {
    int i = omp_get_team_num ();
    int j = omp_get_num_teams ();
    if (j < 417
	|| j > 467
	|| (unsigned) i >= (unsigned) j
	|| a[i] != 3)
      abort ();
    ++a[i];
  }
  e = 4;
  for (k = 0; k < 934; k++)
    {
      if (k >= 417 && k < 467 && a[k] == 3)
	e = 3;
      else if (k == 467)
	e = 2;
      else if (k >= 884 && a[k] == 1)
	e = 1;
      if (a[k] != e)
	abort ();
    }
  return 0;
}