xref: /llvm-project/clang/test/OpenMP/parallel_generic_loop_ast_print.cpp (revision 0c6f2f629cc0017361310fa4c132090413a874db)
1 // RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \
2 // RUN:   -fsyntax-only -verify %s
3 
4 // RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \
5 // RUN:   -ast-print %s | FileCheck %s
6 
7 // RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \
8 // RUN:   -emit-pch -o %t %s
9 
10 // RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \
11 // RUN:   -include-pch %t -ast-print %s | FileCheck %s
12 
13 // expected-no-diagnostics
14 
15 #ifndef HEADER
16 #define HEADER
17 
18 typedef void **omp_allocator_handle_t;
19 extern const omp_allocator_handle_t omp_null_allocator;
20 extern const omp_allocator_handle_t omp_default_mem_alloc;
21 extern const omp_allocator_handle_t omp_large_cap_mem_alloc;
22 extern const omp_allocator_handle_t omp_const_mem_alloc;
23 extern const omp_allocator_handle_t omp_high_bw_mem_alloc;
24 extern const omp_allocator_handle_t omp_low_lat_mem_alloc;
25 extern const omp_allocator_handle_t omp_cgroup_mem_alloc;
26 extern const omp_allocator_handle_t omp_pteam_mem_alloc;
27 extern const omp_allocator_handle_t omp_thread_mem_alloc;
28 
29 //CHECK: template <typename T, int C> void templ_foo(T t) {
30 //CHECK:   T j, z;
31 //CHECK:   #pragma omp parallel loop collapse(C) reduction(+: z) lastprivate(j) bind(thread)
32 //CHECK:   for (T i = 0; i < t; ++i)
33 //CHECK:       for (j = 0; j < t; ++j)
34 //CHECK:           z += i + j;
35 //CHECK: }
36 
37 //CHECK: template<> void templ_foo<int, 2>(int t) {
38 //CHECK:     int j, z;
39 //CHECK:     #pragma omp parallel loop collapse(2) reduction(+: z) lastprivate(j) bind(thread)
40 //CHECK:         for (int i = 0; i < t; ++i)
41 //CHECK:             for (j = 0; j < t; ++j)
42 //CHECK:                 z += i + j;
43 //CHECK: }
44 template <typename T, int C>
templ_foo(T t)45 void templ_foo(T t) {
46 
47   T j,z;
48   #pragma omp parallel loop collapse(C) reduction(+:z) lastprivate(j) bind(thread)
49   for (T i = 0; i<t; ++i)
50     for (j = 0; j<t; ++j)
51       z += i+j;
52 }
53 
54 
55 //CHECK: void test() {
test()56 void test() {
57   constexpr int N = 100;
58   float MTX[N][N];
59   int aaa[1000];
60 
61   //CHECK: #pragma omp target teams distribute parallel for map(tofrom: MTX)
62   //CHECK: #pragma omp parallel loop
63   #pragma omp target teams distribute parallel for map(MTX)
64   for (auto i = 0; i < N; ++i) {
65     #pragma omp parallel loop
66     for (auto j = 0; j < N; ++j) {
67       MTX[i][j] = 0;
68     }
69   }
70 
71   //CHECK: #pragma omp target teams
72   //CHECK: #pragma omp parallel loop
73   #pragma omp target teams
74   for (int i=0; i<1000; ++i) {
75     #pragma omp parallel loop
76     for (int j=0; j<100; j++) {
77       aaa[i] += i + j;
78     }
79   }
80 
81   int j, z, z1, z2 = 1, z3 = 2;
82   //CHECK: #pragma omp parallel loop collapse(2) private(z) lastprivate(j) order(concurrent) firstprivate(z2) num_threads(4) if(1) allocate(omp_default_mem_alloc: z2) shared(z3) reduction(+: z1) bind(parallel) proc_bind(primary)
83   #pragma omp parallel loop collapse(2) private(z) lastprivate(j)          \
84                    order(concurrent) firstprivate(z2) num_threads(4) if(1) \
85                    allocate(omp_default_mem_alloc:z2) shared(z3)           \
86                    reduction(+:z1) bind(parallel) proc_bind(primary)
87   for (auto i = 0; i < N; ++i) {
88     for (j = 0; j < N; ++j) {
89       z = i+j;
90       MTX[i][j] = z;
91       z1 += z;
92     }
93   }
94 
95   //CHECK: #pragma omp target teams
96   //CHECK: #pragma omp parallel loop bind(teams)
97   #pragma omp target teams
98   #pragma omp parallel loop bind(teams)
99   for (auto i = 0; i < N; ++i) { }
100 
101   //CHECK: #pragma omp target
102   //CHECK: #pragma omp teams
103   //CHECK: #pragma omp parallel loop bind(teams)
104   #pragma omp target
105   #pragma omp teams
106   #pragma omp parallel loop bind(teams)
107   for (auto i = 0; i < N; ++i) { }
108 }
109 
110 //CHECK: void nobindingfunc() {
nobindingfunc()111 void nobindingfunc()
112 {
113   //CHECK: #pragma omp parallel loop
114   #pragma omp parallel loop
115   for (int i=0; i<10; ++i) { }
116 }
117 
bar()118 void bar()
119 {
120   templ_foo<int,2>(8);
121 }
122 
123 #endif // HEADER
124