1 // RUN: %libomptarget-compile-and-run-generic
2
3 #include <omp.h>
4 #include <stdio.h>
5 #include <stdlib.h>
6
7 #define NUM_DIMS 3
8
main()9 int main() {
10 int d = omp_get_default_device();
11 int id = omp_get_initial_device();
12 int q[128], q2[128], i;
13 void *p;
14
15 if (d < 0 || d >= omp_get_num_devices())
16 d = id;
17
18 p = omp_target_alloc(130 * sizeof(int), d);
19 if (p == NULL)
20 return 0;
21
22 if (omp_target_memcpy_rect_async(NULL, NULL, 0, 0, NULL, NULL, NULL, NULL,
23 NULL, d, id, 0, NULL) < 3 ||
24 omp_target_memcpy_rect_async(NULL, NULL, 0, 0, NULL, NULL, NULL, NULL,
25 NULL, id, d, 0, NULL) < 3 ||
26 omp_target_memcpy_rect_async(NULL, NULL, 0, 0, NULL, NULL, NULL, NULL,
27 NULL, id, id, 0, NULL) < 3)
28 abort();
29
30 for (i = 0; i < 128; i++)
31 q[i] = 0;
32 if (omp_target_memcpy(p, q, 128 * sizeof(int), 0, 0, d, id) != 0)
33 abort();
34
35 for (i = 0; i < 128; i++)
36 q[i] = i + 1;
37
38 size_t volume[NUM_DIMS] = {1, 2, 3};
39 size_t dst_offsets[NUM_DIMS] = {0, 0, 0};
40 size_t src_offsets[NUM_DIMS] = {0, 0, 0};
41 size_t dst_dimensions[NUM_DIMS] = {3, 4, 5};
42 size_t src_dimensions[NUM_DIMS] = {2, 3, 4};
43
44 if (omp_target_memcpy_rect_async(p, q, sizeof(int), NUM_DIMS, volume,
45 dst_offsets, src_offsets, dst_dimensions,
46 src_dimensions, d, id, 0, NULL) != 0)
47 abort();
48
49 #pragma omp taskwait
50
51 for (i = 0; i < 128; i++)
52 q2[i] = 0;
53 if (omp_target_memcpy(q2, p, 128 * sizeof(int), 0, 0, id, d) != 0)
54 abort();
55
56 /* q2 is expected to contain: 1 2 3 0 0 5 6 7 0 0 .. 0 */
57 if (q2[0] != 1 || q2[1] != 2 || q2[2] != 3 || q2[3] != 0 || q2[4] != 0 ||
58 q2[5] != 5 || q2[6] != 6 || q2[7] != 7)
59 abort();
60 for (i = 8; i < 128; ++i)
61 if (q2[i] != 0)
62 abort();
63
64 omp_target_free(p, d);
65 return 0;
66 }
67