1 | // RUN: %libomptarget-compilexx-generic && \ |
2 | // RUN: env LIBOMPTARGET_STACK_SIZE=2048 %libomptarget-run-generic |
3 | // RUN: %libomptarget-compileoptxx-generic && \ |
4 | // RUN: env LIBOMPTARGET_STACK_SIZE=2048 %libomptarget-run-generic |
5 | |
6 | // We need malloc/global_alloc support |
7 | // UNSUPPORTED: amdgcn-amd-amdhsa |
8 | |
9 | #include <cassert> |
10 | #include <iostream> |
11 | |
12 | void work(int *C) { |
13 | #pragma omp atomic |
14 | ++(*C); |
15 | } |
16 | |
17 | void use(int *C) { |
18 | #pragma omp parallel num_threads(2) |
19 | work(C); |
20 | } |
21 | |
22 | int main() { |
23 | int C = 0; |
24 | #pragma omp target map(C) |
25 | { |
26 | use(C: &C); |
27 | #pragma omp parallel num_threads(2) |
28 | use(C: &C); |
29 | } |
30 | |
31 | assert(C >= 2 && C <= 6); |
32 | |
33 | std::cout << "PASS\n" ; |
34 | |
35 | return 0; |
36 | } |
37 | |
38 | // CHECK: PASS |
39 | |