xref: /llvm-project/offload/test/mapping/declare_mapper_target_update.cpp (revision 330d8983d25d08580fc1642fea48b2473f47a9da)
1 // RUN: %libomptarget-compile-run-and-check-generic
2 
3 #include <cstdio>
4 #include <cstdlib>
5 
6 #define NUM 1024
7 
8 class C {
9 public:
10   int *a;
11 };
12 
13 #pragma omp declare mapper(id : C s) map(s.a[0 : NUM])
14 
main()15 int main() {
16   C c;
17   int sum = 0;
18   c.a = (int *)malloc(sizeof(int) * NUM);
19   for (int i = 0; i < NUM; i++) {
20     c.a[i] = 1;
21   }
22 #pragma omp target enter data map(mapper(id), alloc : c)
23 #pragma omp target teams distribute parallel for
24   for (int i = 0; i < NUM; i++) {
25     c.a[i] = 0;
26   }
27 #pragma omp target update from(mapper(id) : c)
28   for (int i = 0; i < NUM; i++) {
29     sum += c.a[i];
30   }
31   // CHECK: Sum (after first update from) = 0
32   printf("Sum (after first update from) = %d\n", sum);
33   for (int i = 0; i < NUM; i++) {
34     c.a[i] = 1;
35   }
36 #pragma omp target update to(mapper(id) : c)
37 #pragma omp target teams distribute parallel for
38   for (int i = 0; i < NUM; i++) {
39     ++c.a[i];
40   }
41   sum = 0;
42   for (int i = 0; i < NUM; i++) {
43     sum += c.a[i];
44   }
45   // CHECK: Sum (after update to) = 1024
46   printf("Sum (after update to) = %d\n", sum);
47 #pragma omp target update from(mapper(id) : c)
48   sum = 0;
49   for (int i = 0; i < NUM; i++) {
50     sum += c.a[i];
51   }
52   // CHECK: Sum (after second update from) = 2048
53   printf("Sum (after second update from) = %d\n", sum);
54 #pragma omp target exit data map(mapper(id), delete : c)
55   return 0;
56 }
57