1*330d8983SJohannes Doerfert // RUN: %libomptarget-compile-and-run-generic
2*330d8983SJohannes Doerfert
3*330d8983SJohannes Doerfert #include "stdio.h"
4*330d8983SJohannes Doerfert #include <omp.h>
5*330d8983SJohannes Doerfert #include <stdlib.h>
6*330d8983SJohannes Doerfert
main()7*330d8983SJohannes Doerfert int main() {
8*330d8983SJohannes Doerfert int d = omp_get_default_device();
9*330d8983SJohannes Doerfert int id = omp_get_initial_device();
10*330d8983SJohannes Doerfert int q[128], i;
11*330d8983SJohannes Doerfert void *p;
12*330d8983SJohannes Doerfert void *result;
13*330d8983SJohannes Doerfert
14*330d8983SJohannes Doerfert if (d < 0 || d >= omp_get_num_devices())
15*330d8983SJohannes Doerfert d = id;
16*330d8983SJohannes Doerfert
17*330d8983SJohannes Doerfert p = omp_target_alloc(130 * sizeof(int), d);
18*330d8983SJohannes Doerfert if (p == NULL)
19*330d8983SJohannes Doerfert return 0;
20*330d8983SJohannes Doerfert
21*330d8983SJohannes Doerfert for (i = 0; i < 128; i++)
22*330d8983SJohannes Doerfert q[i] = i;
23*330d8983SJohannes Doerfert
24*330d8983SJohannes Doerfert result = omp_target_memset(p, 0, 130 * sizeof(int), d);
25*330d8983SJohannes Doerfert if (result != p) {
26*330d8983SJohannes Doerfert abort();
27*330d8983SJohannes Doerfert }
28*330d8983SJohannes Doerfert
29*330d8983SJohannes Doerfert int q2[128];
30*330d8983SJohannes Doerfert for (i = 0; i < 128; ++i)
31*330d8983SJohannes Doerfert q2[i] = i;
32*330d8983SJohannes Doerfert if (omp_target_memcpy_async(q2, p, 128 * sizeof(int), 0, sizeof(int), id, d,
33*330d8983SJohannes Doerfert 0, NULL))
34*330d8983SJohannes Doerfert abort();
35*330d8983SJohannes Doerfert
36*330d8983SJohannes Doerfert #pragma omp taskwait
37*330d8983SJohannes Doerfert
38*330d8983SJohannes Doerfert for (i = 0; i < 128; ++i)
39*330d8983SJohannes Doerfert if (q2[i] != 0)
40*330d8983SJohannes Doerfert abort();
41*330d8983SJohannes Doerfert
42*330d8983SJohannes Doerfert omp_target_free(p, d);
43*330d8983SJohannes Doerfert
44*330d8983SJohannes Doerfert return 0;
45*330d8983SJohannes Doerfert }
46