aboutsummaryrefslogtreecommitdiff
path: root/libgomp/testsuite/libgomp.c-c++-common/get-mapped-ptr-2.c
blob: 194dade8ac5f75cb8fa3ec9dccb9142bf8dbebd5 (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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <omp.h>
#include <stdlib.h>
#include <stdint.h>

int
main ()
{
  int d = omp_get_default_device ();
  int id = omp_get_initial_device ();
  int a = 42;
  int b[] = { 24, 42 };
  int c[] = { 47, 11 };
  int e[128];
  int *q = &a;
  void *p1 = NULL, *p2 = NULL, *p3 = NULL;
  void *devptrs[128];

  if (d < 0 || d >= omp_get_num_devices ())
    d = id;

  for (int i = 0; i < 128; i++)
    e[i] = i;

  #pragma omp target data map(alloc: a, b, c[1], e[32:64]) device(d)
  {
    #pragma omp target map(from: p1, p2, p3, devptrs) map(alloc: a, b, c[1], e[32:64]) device(d)
    {
      p1 = &a;
      p2 = &b;
      p3 = &c[1];
      for (int i = 32; i < 96; i++)
	devptrs[i] = &e[i];
    }

    if (omp_get_mapped_ptr (&a, d) != (d == id ? &a : p1)
	|| omp_get_mapped_ptr (q, d) != (d == id ? q : p1)
	|| omp_get_mapped_ptr (b, d) != (d == id ? b : p2)
	|| omp_get_mapped_ptr (&b[0], d) != (d == id ? &b[0] : p2)
	|| omp_get_mapped_ptr (&c[1], d) != (d == id ? &c[1] : p3)
	|| omp_get_mapped_ptr (&c[0], d) != (d == id ? &c[0] : NULL))
      abort ();

    for (int i = 0; i < 32; i++)
      if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : NULL))
	abort ();
    for (int i = 32; i < 96; i++)
      if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : devptrs[i]))
	abort ();
    for (int i = 96; i < 128; i++)
      if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : NULL))
	abort ();
  }

  if (omp_get_mapped_ptr (&a, d) != (d == id ? &a : NULL)
      || omp_get_mapped_ptr (q, d) != (d == id ? q : NULL)
      || omp_get_mapped_ptr (b, d) != (d == id ? b : NULL)
      || omp_get_mapped_ptr (&b[0], d) != (d == id ? &b[0] : NULL)
      || omp_get_mapped_ptr (&c[1], d) != (d == id ? &c[1] : NULL)
      || omp_get_mapped_ptr (&c[0], d) != (d == id ? &c[0] : NULL))
    abort ();
  for (int i = 0; i < 128; i++)
    if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : NULL))
      abort ();

  #pragma omp target enter data map (alloc: a, b, c[1], e[32:64]) device (d)
  #pragma omp target map(from: p1, p2, p3, devptrs) map(alloc: a, b, c[1], e[32:64]) device(d)
  {
    p1 = &a;
    p2 = &b;
    p3 = &c[1];
    for (int i = 32; i < 96; i++)
      devptrs[i] = &e[i];
  }

  if (omp_get_mapped_ptr (&a, d) != (d == id ? &a : p1)
      || omp_get_mapped_ptr (q, d) != (d == id ? q : p1)
      || omp_get_mapped_ptr (b, d) != (d == id ? b : p2)
      || omp_get_mapped_ptr (&b[0], d) != (d == id ? &b[0] : p2)
      || omp_get_mapped_ptr (&c[1], d) != (d == id ? &c[1] : p3)
      || omp_get_mapped_ptr (&c[0], d) != (d == id ? &c[0] : NULL))
    abort ();
  for (int i = 0; i < 32; i++)
    if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : NULL))
      abort ();
  for (int i = 32; i < 96; i++)
    if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : devptrs[i]))
      abort ();
  for (int i = 96; i < 128; i++)
    if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : NULL))
      abort ();

  #pragma omp target exit data map (delete: a, b, c[1], e[32:64]) device (d)

  if (omp_get_mapped_ptr (&a, d) != (d == id ? &a : NULL)
      || omp_get_mapped_ptr (q, d) != (d == id ? q : NULL)
      || omp_get_mapped_ptr (b, d) != (d == id ? b : NULL)
      || omp_get_mapped_ptr (&b[0], d) != (d == id ? &b[0] : NULL)
      || omp_get_mapped_ptr (&c[1], d) != (d == id ? &c[1] : NULL)
      || omp_get_mapped_ptr (&c[0], d) != (d == id ? &c[0] : NULL))
    abort ();
  for (int i = 0; i < 128; i++)
    if (omp_get_mapped_ptr (&e[i], d) != (d == id ? &e[i] : NULL))
      abort ();

  return 0;
}