74 lines · c
1// RUN: %libomptarget-compile-and-run-generic2 3#include "stdio.h"4#include <omp.h>5#include <stdlib.h>6 7int main() {8 int d = omp_get_default_device();9 int id = omp_get_initial_device();10 int a[128], b[64], c[32], e[16], q[128], i;11 void *p;12 13 if (d < 0 || d >= omp_get_num_devices())14 d = id;15 16 p = omp_target_alloc(130 * sizeof(int), d);17 if (p == NULL)18 return 0;19 20 for (i = 0; i < 128; ++i)21 a[i] = i + 1;22 for (i = 0; i < 64; ++i)23 b[i] = i + 2;24 for (i = 0; i < 32; i++)25 c[i] = 0;26 for (i = 0; i < 16; i++)27 e[i] = i + 4;28 29 omp_depend_t obj[2];30 31#pragma omp parallel num_threads(5)32#pragma omp single33 {34#pragma omp task depend(out : p)35 omp_target_memcpy(p, a, 128 * sizeof(int), 0, 0, d, id);36 37#pragma omp task depend(inout : p)38 omp_target_memcpy(p, b, 64 * sizeof(int), 0, 0, d, id);39 40#pragma omp task depend(out : c)41 for (i = 0; i < 32; i++)42 c[i] = i + 3;43 44#pragma omp depobj(obj[0]) depend(inout : p)45#pragma omp depobj(obj[1]) depend(in : c)46 omp_target_memcpy_async(p, c, 32 * sizeof(int), 0, 0, d, id, 2, obj);47 48#pragma omp task depend(in : p)49 omp_target_memcpy(p, e, 16 * sizeof(int), 0, 0, d, id);50 }51 52#pragma omp taskwait53 54 for (i = 0; i < 128; ++i)55 q[i] = 0;56 omp_target_memcpy(q, p, 128 * sizeof(int), 0, 0, id, d);57 for (i = 0; i < 16; ++i)58 if (q[i] != i + 4)59 abort();60 for (i = 16; i < 32; ++i)61 if (q[i] != i + 3)62 abort();63 for (i = 32; i < 64; ++i)64 if (q[i] != i + 2)65 abort();66 for (i = 64; i < 128; ++i)67 if (q[i] != i + 1)68 abort();69 70 omp_target_free(p, d);71 72 return 0;73}74