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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#include <omp.h>
#include <stdlib.h>
int
main ()
{
int d = omp_get_default_device ();
int id = omp_get_initial_device ();
int err;
int q[128], i;
void *p;
if (d < 0 || d >= omp_get_num_devices ())
d = id;
for (i = 0; i < 128; i++)
q[i] = i;
p = omp_target_alloc (130 * sizeof (int), d);
if (p == NULL)
return 0;
if (omp_target_memcpy_rect (NULL, NULL, 0, 0, NULL, NULL, NULL, NULL, NULL,
d, id) < 3
|| omp_target_memcpy_rect (NULL, NULL, 0, 0, NULL, NULL, NULL, NULL,
NULL, id, d) < 3
|| omp_target_memcpy_rect (NULL, NULL, 0, 0, NULL, NULL, NULL, NULL,
NULL, id, id) < 3)
abort ();
if (omp_target_associate_ptr (q, p, 128 * sizeof (int), sizeof (int), d) == 0)
{
size_t volume[3] = { 128, 0, 0 };
size_t dst_offsets[3] = { 0, 0, 0 };
size_t src_offsets[3] = { 1, 0, 0 };
size_t dst_dimensions[3] = { 128, 0, 0 };
size_t src_dimensions[3] = { 128, 0, 0 };
if (omp_target_associate_ptr (q, p, 128 * sizeof (int), sizeof (int), d) != 0)
abort ();
if (omp_target_is_present (q, d) != 1
|| omp_target_is_present (&q[32], d) != 1
|| omp_target_is_present (&q[128], d) != 1)
abort ();
if (omp_target_memcpy (p, q, 128 * sizeof (int), sizeof (int), 0,
d, id) != 0)
abort ();
#pragma omp target if (d >= 0) device (d >= 0 ? d : 0) map(alloc:q[0:32]) map(from:err)
{
int j;
err = 0;
for (j = 0; j < 128; j++)
if (q[j] != j)
err = 1;
else
q[j] += 4;
}
if (err)
abort ();
if (omp_target_memcpy_rect (q, p, sizeof (int), 1, volume,
dst_offsets, src_offsets, dst_dimensions,
src_dimensions, id, d) != 0)
abort ();
for (i = 0; i < 128; i++)
if (q[i] != i + 4)
abort ();
volume[2] = 2;
volume[1] = 3;
volume[0] = 6;
dst_offsets[2] = 1;
dst_offsets[1] = 0;
dst_offsets[0] = 0;
src_offsets[2] = 1;
src_offsets[1] = 0;
src_offsets[0] = 3;
dst_dimensions[2] = 2;
dst_dimensions[1] = 3;
dst_dimensions[0] = 6;
src_dimensions[2] = 3;
src_dimensions[1] = 4;
src_dimensions[0] = 6;
if (omp_target_memcpy_rect (p, q, sizeof (int), 3, volume,
dst_offsets, src_offsets, dst_dimensions,
src_dimensions, d, id) != 0)
abort ();
#pragma omp target if (d >= 0) device (d >= 0 ? d : 0) map(alloc:q[0:32]) map(from:err)
{
int j, k, l;
err = 0;
for (j = 0; j < 6; j++)
for (k = 0; k < 3; k++)
for (l = 0; l < 2; l++)
if (q[j * 6 + k * 2 + l] != 3 * 12 + 4 + 1 + l + k * 3 + j * 12)
err = 1;
}
if (err)
abort ();
if (omp_target_memcpy (p, p, 10 * sizeof (int), 51 * sizeof (int),
111 * sizeof (int), d, d) != 0)
abort ();
#pragma omp target if (d >= 0) device (d >= 0 ? d : 0) map(alloc:q[0:32]) map(from:err)
{
int j;
err = 0;
for (j = 0; j < 10; j++)
if (q[50 + j] != q[110 + j])
err = 1;
}
if (err)
abort ();
if (omp_target_disassociate_ptr (q, d) != 0)
abort ();
}
omp_target_free (p, d);
return 0;
}
|
