1// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple x86_64-unknown-unknown -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s
2// RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
3// RUN: %clang_cc1 -fopenmp -x c++ -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
4// REQUIRES: x86-registered-target
5// expected-no-diagnostics
6#ifndef HEADER
7#define HEADER
8
9// CHECK: [[IDENT_T_TY:%.+]] = type { i32, i32, i32, i32, i8* }
10// CHECK: [[IMPLICIT_BARRIER_LOC:@.+]] = private unnamed_addr constant %{{.+}} { i32 0, i32 66, i32 0, i32 0, i8*
11// CHECK-LABEL: define {{.*void}} @{{.*}}static_not_chunked{{.*}}(float* {{.+}}, float* {{.+}}, float* {{.+}}, float* {{.+}})
12void static_not_chunked(float *a, float *b, float *c, float *d) {
13// CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num([[IDENT_T_TY]]* [[DEFAULT_LOC:[@%].+]])
14  #pragma omp for schedule(static) ordered
15// CHECK: call void @__kmpc_dispatch_init_4([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32 66, i32 0, i32 4571423, i32 1, i32 1)
16//
17// CHECK: [[HASWORK:%.+]] = call i32 @__kmpc_dispatch_next_4([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32* [[OMP_ISLAST:%[^,]+]], i32* [[OMP_LB:%[^,]+]], i32* [[OMP_UB:%[^,]+]], i32* [[OMP_ST:%[^,]+]])
18// CHECK-NEXT: [[O_CMP:%.+]] = icmp ne i32 [[HASWORK]], 0
19// CHECK-NEXT: br i1 [[O_CMP]], label %[[O_LOOP1_BODY:[^,]+]], label %[[O_LOOP1_END:[^,]+]]
20
21// Loop header
22// CHECK: [[O_LOOP1_BODY]]
23// CHECK: [[LB:%.+]] = load i32, i32* [[OMP_LB]]
24// CHECK-NEXT: store i32 [[LB]], i32* [[OMP_IV:[^,]+]]
25// CHECK: [[IV:%.+]] = load i32, i32* [[OMP_IV]]
26
27// CHECK-NEXT: [[UB:%.+]] = load i32, i32* [[OMP_UB]]
28// CHECK-NEXT: [[CMP:%.+]] = icmp sle i32 [[IV]], [[UB]]
29// CHECK-NEXT: br i1 [[CMP]], label %[[LOOP1_BODY:[^,]+]], label %[[LOOP1_END:[^,]+]]
30  for (int i = 32000000; i > 33; i += -7) {
31// CHECK: [[LOOP1_BODY]]
32// Start of body: calculate i from IV:
33// CHECK: [[IV1_1:%.+]] = load i32, i32* [[OMP_IV]]
34// CHECK-NEXT: [[CALC_I_1:%.+]] = mul nsw i32 [[IV1_1]], 7
35// CHECK-NEXT: [[CALC_I_2:%.+]] = sub nsw i32 32000000, [[CALC_I_1]]
36// CHECK-NEXT: store i32 [[CALC_I_2]], i32* [[LC_I:.+]]
37
38// ... start of ordered region ...
39// CHECK-NEXT: call void @__kmpc_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
40// ... loop body ...
41// End of body: store into a[i]:
42// CHECK: store float [[RESULT:%.+]], float* {{%.+}}
43// CHECK-NOT: !llvm.mem.parallel_loop_access
44// CHECK-NEXT: call void @__kmpc_end_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
45// ... end of ordered region ...
46    #pragma omp ordered
47    a[i] = b[i] * c[i] * d[i];
48// CHECK: [[IV1_2:%.+]] = load i32, i32* [[OMP_IV]]{{.*}}
49// CHECK-NEXT: [[ADD1_2:%.+]] = add nsw i32 [[IV1_2]], 1
50// CHECK-NEXT: store i32 [[ADD1_2]], i32* [[OMP_IV]]
51// CHECK-NEXT: call void @__kmpc_dispatch_fini_4([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
52// CHECK-NEXT: br label %{{.+}}
53  }
54// CHECK: [[LOOP1_END]]
55// CHECK: [[O_LOOP1_END]]
56// CHECK: call {{.+}} @__kmpc_barrier([[IDENT_T_TY]]* [[IMPLICIT_BARRIER_LOC]], i32 [[GTID]])
57// CHECK: ret void
58}
59
60// CHECK-LABEL: define {{.*void}} @{{.*}}dynamic1{{.*}}(float* {{.+}}, float* {{.+}}, float* {{.+}}, float* {{.+}})
61void dynamic1(float *a, float *b, float *c, float *d) {
62// CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num([[IDENT_T_TY]]* [[DEFAULT_LOC:[@%].+]])
63  #pragma omp for schedule(dynamic) ordered
64// CHECK: call void @__kmpc_dispatch_init_8u([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32 67, i64 0, i64 16908287, i64 1, i64 1)
65//
66// CHECK: [[HASWORK:%.+]] = call i32 @__kmpc_dispatch_next_8u([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32* [[OMP_ISLAST:%[^,]+]], i64* [[OMP_LB:%[^,]+]], i64* [[OMP_UB:%[^,]+]], i64* [[OMP_ST:%[^,]+]])
67// CHECK-NEXT: [[O_CMP:%.+]] = icmp ne i32 [[HASWORK]], 0
68// CHECK-NEXT: br i1 [[O_CMP]], label %[[O_LOOP1_BODY:[^,]+]], label %[[O_LOOP1_END:[^,]+]]
69
70// Loop header
71// CHECK: [[O_LOOP1_BODY]]
72// CHECK: [[LB:%.+]] = load i64, i64* [[OMP_LB]]
73// CHECK-NEXT: store i64 [[LB]], i64* [[OMP_IV:[^,]+]]
74// CHECK: [[IV:%.+]] = load i64, i64* [[OMP_IV]]
75
76// CHECK-NEXT: [[UB:%.+]] = load i64, i64* [[OMP_UB]]
77// CHECK-NEXT: [[CMP:%.+]] = icmp ule i64 [[IV]], [[UB]]
78// CHECK-NEXT: br i1 [[CMP]], label %[[LOOP1_BODY:[^,]+]], label %[[LOOP1_END:[^,]+]]
79  for (unsigned long long i = 131071; i < 2147483647; i += 127) {
80// CHECK: [[LOOP1_BODY]]
81// Start of body: calculate i from IV:
82// CHECK: [[IV1_1:%.+]] = load i64, i64* [[OMP_IV]]
83// CHECK-NEXT: [[CALC_I_1:%.+]] = mul i64 [[IV1_1]], 127
84// CHECK-NEXT: [[CALC_I_2:%.+]] = add i64 131071, [[CALC_I_1]]
85// CHECK-NEXT: store i64 [[CALC_I_2]], i64* [[LC_I:.+]]
86
87// ... start of ordered region ...
88// CHECK-NEXT: call void @__kmpc_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
89// ... loop body ...
90// End of body: store into a[i]:
91// CHECK: store float [[RESULT:%.+]], float* {{%.+}}
92// CHECK-NOT: !llvm.mem.parallel_loop_access
93// CHECK-NEXT: call void @__kmpc_end_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
94// ... end of ordered region ...
95    #pragma omp ordered threads
96    a[i] = b[i] * c[i] * d[i];
97// CHECK: [[IV1_2:%.+]] = load i64, i64* [[OMP_IV]]{{.*}}
98// CHECK-NEXT: [[ADD1_2:%.+]] = add i64 [[IV1_2]], 1
99// CHECK-NEXT: store i64 [[ADD1_2]], i64* [[OMP_IV]]
100
101// ... end iteration for ordered loop ...
102// CHECK-NEXT: call void @__kmpc_dispatch_fini_8u([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
103// CHECK-NEXT: br label %{{.+}}
104  }
105// CHECK: [[LOOP1_END]]
106// CHECK: [[O_LOOP1_END]]
107// CHECK: call {{.+}} @__kmpc_barrier([[IDENT_T_TY]]* [[IMPLICIT_BARRIER_LOC]], i32 [[GTID]])
108// CHECK: ret void
109}
110
111// CHECK-LABEL: define {{.*void}} @{{.*}}test_auto{{.*}}(float* {{.+}}, float* {{.+}}, float* {{.+}}, float* {{.+}})
112void test_auto(float *a, float *b, float *c, float *d) {
113  unsigned int x = 0;
114  unsigned int y = 0;
115// CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num([[IDENT_T_TY]]* [[DEFAULT_LOC:[@%].+]])
116  #pragma omp for schedule(auto) collapse(2) ordered
117// CHECK: call void @__kmpc_dispatch_init_8([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32 70, i64 0, i64 [[LAST_ITER:%[^,]+]], i64 1, i64 1)
118//
119// CHECK: [[HASWORK:%.+]] = call i32 @__kmpc_dispatch_next_8([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32* [[OMP_ISLAST:%[^,]+]], i64* [[OMP_LB:%[^,]+]], i64* [[OMP_UB:%[^,]+]], i64* [[OMP_ST:%[^,]+]])
120// CHECK-NEXT: [[O_CMP:%.+]] = icmp ne i32 [[HASWORK]], 0
121// CHECK-NEXT: br i1 [[O_CMP]], label %[[O_LOOP1_BODY:[^,]+]], label %[[O_LOOP1_END:[^,]+]]
122
123// Loop header
124// CHECK: [[O_LOOP1_BODY]]
125// CHECK: [[LB:%.+]] = load i64, i64* [[OMP_LB]]
126// CHECK-NEXT: store i64 [[LB]], i64* [[OMP_IV:[^,]+]]
127// CHECK: [[IV:%.+]] = load i64, i64* [[OMP_IV]]
128
129// CHECK-NEXT: [[UB:%.+]] = load i64, i64* [[OMP_UB]]
130// CHECK-NEXT: [[CMP:%.+]] = icmp sle i64 [[IV]], [[UB]]
131// CHECK-NEXT: br i1 [[CMP]], label %[[LOOP1_BODY:[^,]+]], label %[[LOOP1_END:[^,]+]]
132// FIXME: When the iteration count of some nested loop is not a known constant,
133// we should pre-calculate it, like we do for the total number of iterations!
134  for (char i = static_cast<char>(y); i <= '9'; ++i)
135    for (x = 11; x > 0; --x) {
136// CHECK: [[LOOP1_BODY]]
137// Start of body: indices are calculated from IV:
138// CHECK: store i8 {{%[^,]+}}, i8* {{%[^,]+}}
139// CHECK: store i32 {{%[^,]+}}, i32* {{%[^,]+}}
140
141// ... start of ordered region ...
142// CHECK: call void @__kmpc_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
143// ... loop body ...
144// End of body: store into a[i]:
145// CHECK: store float [[RESULT:%.+]], float* {{%.+}}
146// CHECK-NOT: !llvm.mem.parallel_loop_access
147// CHECK-NEXT: call void @__kmpc_end_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
148// ... end of ordered region ...
149    #pragma omp ordered
150    a[i] = b[i] * c[i] * d[i];
151// CHECK: [[IV1_2:%.+]] = load i64, i64* [[OMP_IV]]{{.*}}
152// CHECK-NEXT: [[ADD1_2:%.+]] = add nsw i64 [[IV1_2]], 1
153// CHECK-NEXT: store i64 [[ADD1_2]], i64* [[OMP_IV]]
154
155// ... end iteration for ordered loop ...
156// CHECK-NEXT: call void @__kmpc_dispatch_fini_8([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
157// CHECK-NEXT: br label %{{.+}}
158  }
159// CHECK: [[LOOP1_END]]
160// CHECK: [[O_LOOP1_END]]
161// CHECK: call {{.+}} @__kmpc_barrier([[IDENT_T_TY]]* [[IMPLICIT_BARRIER_LOC]], i32 [[GTID]])
162// CHECK: ret void
163}
164
165// CHECK-LABEL: define {{.*void}} @{{.*}}runtime{{.*}}(float* {{.+}}, float* {{.+}}, float* {{.+}}, float* {{.+}})
166void runtime(float *a, float *b, float *c, float *d) {
167  int x = 0;
168// CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num([[IDENT_T_TY]]* [[DEFAULT_LOC:[@%].+]])
169  #pragma omp for collapse(2) schedule(runtime) ordered
170// CHECK: call void @__kmpc_dispatch_init_4([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32 69, i32 0, i32 199, i32 1, i32 1)
171//
172// CHECK: [[HASWORK:%.+]] = call i32 @__kmpc_dispatch_next_4([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]], i32* [[OMP_ISLAST:%[^,]+]], i32* [[OMP_LB:%[^,]+]], i32* [[OMP_UB:%[^,]+]], i32* [[OMP_ST:%[^,]+]])
173// CHECK-NEXT: [[O_CMP:%.+]] = icmp ne i32 [[HASWORK]], 0
174// CHECK-NEXT: br i1 [[O_CMP]], label %[[O_LOOP1_BODY:[^,]+]], label %[[O_LOOP1_END:[^,]+]]
175
176// Loop header
177// CHECK: [[O_LOOP1_BODY]]
178// CHECK: [[LB:%.+]] = load i32, i32* [[OMP_LB]]
179// CHECK-NEXT: store i32 [[LB]], i32* [[OMP_IV:[^,]+]]
180// CHECK: [[IV:%.+]] = load i32, i32* [[OMP_IV]]
181
182// CHECK-NEXT: [[UB:%.+]] = load i32, i32* [[OMP_UB]]
183// CHECK-NEXT: [[CMP:%.+]] = icmp sle i32 [[IV]], [[UB]]
184// CHECK-NEXT: br i1 [[CMP]], label %[[LOOP1_BODY:[^,]+]], label %[[LOOP1_END:[^,]+]]
185  for (unsigned char i = '0' ; i <= '9'; ++i)
186    for (x = -10; x < 10; ++x) {
187// CHECK: [[LOOP1_BODY]]
188// Start of body: indices are calculated from IV:
189// CHECK: store i8 {{%[^,]+}}, i8* {{%[^,]+}}
190// CHECK: store i32 {{%[^,]+}}, i32* {{%[^,]+}}
191
192// ... start of ordered region ...
193// CHECK: call void @__kmpc_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
194// ... loop body ...
195// End of body: store into a[i]:
196// CHECK: store float [[RESULT:%.+]], float* {{%.+}}
197// CHECK-NOT: !llvm.mem.parallel_loop_access
198// CHECK-NEXT: call void @__kmpc_end_ordered([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
199// ... end of ordered region ...
200    #pragma omp ordered threads
201    a[i] = b[i] * c[i] * d[i];
202// CHECK: [[IV1_2:%.+]] = load i32, i32* [[OMP_IV]]{{.*}}
203// CHECK-NEXT: [[ADD1_2:%.+]] = add nsw i32 [[IV1_2]], 1
204// CHECK-NEXT: store i32 [[ADD1_2]], i32* [[OMP_IV]]
205
206// ... end iteration for ordered loop ...
207// CHECK-NEXT: call void @__kmpc_dispatch_fini_4([[IDENT_T_TY]]* [[DEFAULT_LOC]], i32 [[GTID]])
208// CHECK-NEXT: br label %{{.+}}
209  }
210// CHECK: [[LOOP1_END]]
211// CHECK: [[O_LOOP1_END]]
212// CHECK: call {{.+}} @__kmpc_barrier([[IDENT_T_TY]]* [[IMPLICIT_BARRIER_LOC]], i32 [[GTID]])
213// CHECK: ret void
214}
215
216float f[10];
217// CHECK-LABEL: foo_simd
218void foo_simd(int low, int up) {
219  // CHECK: store float 0.000000e+00, float* %{{.+}}, align {{[0-9]+}}, !llvm.mem.parallel_loop_access !
220  // CHECK-NEXT: call void [[CAP_FUNC:@.+]](i32* %{{.+}}) #{{[0-9]+}}, !llvm.mem.parallel_loop_access !
221#pragma omp simd
222  for (int i = low; i < up; ++i) {
223    f[i] = 0.0;
224#pragma omp ordered simd
225    f[i] = 1.0;
226  }
227}
228
229// CHECK: define internal void [[CAP_FUNC]](i32* dereferenceable({{[0-9]+}}) %{{.+}}) #
230// CHECK: store float 1.000000e+00, float* %{{.+}}, align
231// CHECK-NEXT: ret void
232
233#endif // HEADER
234
235