1*34ee2549SAlexey Bataev // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=50 -x c++ -triple x86_64-unknown-unknown -emit-llvm %s -o - | FileCheck %s
2*34ee2549SAlexey Bataev // RUN: %clang_cc1 -fopenmp -fopenmp-version=50 -x c++ -std=c++11 -triple x86_64-unknown-unknown -emit-pch -o %t %s
3*34ee2549SAlexey Bataev // RUN: %clang_cc1 -fopenmp -fopenmp-version=50 -x c++ -triple x86_64-unknown-unknown -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
4*34ee2549SAlexey Bataev 
5*34ee2549SAlexey Bataev // RUN: %clang_cc1 -verify -fopenmp-simd -fopenmp-version=50 -x c++ -triple x86_64-unknown-unknown -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
6*34ee2549SAlexey Bataev // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=50 -x c++ -std=c++11 -triple x86_64-unknown-unknown -emit-pch -o %t %s
7*34ee2549SAlexey Bataev // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=50 -x c++ -triple x86_64-unknown-unknown -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
8*34ee2549SAlexey Bataev // SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
9*34ee2549SAlexey Bataev // expected-no-diagnostics
10*34ee2549SAlexey Bataev #ifndef HEADER
11*34ee2549SAlexey Bataev #define HEADER
12*34ee2549SAlexey Bataev 
13*34ee2549SAlexey Bataev void foo();
14*34ee2549SAlexey Bataev void bar();
15*34ee2549SAlexey Bataev 
16*34ee2549SAlexey Bataev // CHECK: define void @{{.*}}baz{{.*}}(i32 %n)
17*34ee2549SAlexey Bataev void baz(int n) {
18*34ee2549SAlexey Bataev   static float a[10];
19*34ee2549SAlexey Bataev   static double b;
20*34ee2549SAlexey Bataev   // CHECK: call i8* @llvm.stacksave()
21*34ee2549SAlexey Bataev   // CHECK: [[A_BUF_SIZE:%.+]] = mul nuw i64 10, [[NUM_ELEMS:%[^,]+]]
22*34ee2549SAlexey Bataev 
23*34ee2549SAlexey Bataev   // float a_buffer[10][n];
24*34ee2549SAlexey Bataev   // CHECK: [[A_BUF:%.+]] = alloca float, i64 [[A_BUF_SIZE]],
25*34ee2549SAlexey Bataev 
26*34ee2549SAlexey Bataev   // double b_buffer[10];
27*34ee2549SAlexey Bataev   // CHECK: [[B_BUF:%.+]] = alloca double, i64 10,
28*34ee2549SAlexey Bataev #pragma omp for simd reduction(inscan, +:a[:n], b)
29*34ee2549SAlexey Bataev   for (int i = 0; i < 10; ++i) {
30*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_init_4(
31*34ee2549SAlexey Bataev     // CHECK: call i8* @llvm.stacksave()
32*34ee2549SAlexey Bataev     // CHECK: store float 0.000000e+00, float* %
33*34ee2549SAlexey Bataev     // CHECK: store double 0.000000e+00, double* [[B_PRIV_ADDR:%.+]],
34*34ee2549SAlexey Bataev     // CHECK: br label %[[DISPATCH:[^,]+]]
35*34ee2549SAlexey Bataev     // CHECK: [[INPUT_PHASE:.+]]:
36*34ee2549SAlexey Bataev     // CHECK: call void @{{.+}}foo{{.+}}()
37*34ee2549SAlexey Bataev 
38*34ee2549SAlexey Bataev     // a_buffer[i][0..n] = a_priv[[0..n];
39*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX_I:%.+]] = load i32, i32* [[IV_ADDR:%.+]],
40*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX:%.+]] = zext i32 [[BASE_IDX_I]] to i64
41*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[BASE_IDX]], [[NUM_ELEMS]]
42*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
43*34ee2549SAlexey Bataev     // CHECK: [[A_PRIV:%.+]] = getelementptr inbounds [10 x float], [10 x float]* [[A_PRIV_ADDR:%.+]], i64 0, i64 0
44*34ee2549SAlexey Bataev     // CHECK: [[BYTES:%.+]] = mul nuw i64 [[NUM_ELEMS:%.+]], 4
45*34ee2549SAlexey Bataev     // CHECK: [[DEST:%.+]] = bitcast float* [[A_BUF_IDX]] to i8*
46*34ee2549SAlexey Bataev     // CHECK: [[SRC:%.+]] = bitcast float* [[A_PRIV]] to i8*
47*34ee2549SAlexey Bataev     // CHECK: call void @llvm.memcpy.p0i8.p0i8.i64(i8* {{.*}}[[DEST]], i8* {{.*}}[[SRC]], i64 [[BYTES]], i1 false)
48*34ee2549SAlexey Bataev 
49*34ee2549SAlexey Bataev     // b_buffer[i] = b_priv;
50*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[BASE_IDX]]
51*34ee2549SAlexey Bataev     // CHECK: [[B_PRIV:%.+]] = load double, double* [[B_PRIV_ADDR]],
52*34ee2549SAlexey Bataev     // CHECK: store double [[B_PRIV]], double* [[B_BUF_IDX]],
53*34ee2549SAlexey Bataev     // CHECK: br label %[[LOOP_CONTINUE:.+]]
54*34ee2549SAlexey Bataev 
55*34ee2549SAlexey Bataev     // CHECK: [[DISPATCH]]:
56*34ee2549SAlexey Bataev     // CHECK: br label %[[INPUT_PHASE]]
57*34ee2549SAlexey Bataev     // CHECK: [[LOOP_CONTINUE]]:
58*34ee2549SAlexey Bataev     // CHECK: call void @llvm.stackrestore(i8* %
59*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_fini(
60*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_barrier(
61*34ee2549SAlexey Bataev     foo();
62*34ee2549SAlexey Bataev #pragma omp scan inclusive(a[:n], b)
63*34ee2549SAlexey Bataev     // CHECK: [[LOG2_10:%.+]] = call double @llvm.log2.f64(double 1.000000e+01)
64*34ee2549SAlexey Bataev     // CHECK: [[CEIL_LOG2_10:%.+]] = call double @llvm.ceil.f64(double [[LOG2_10]])
65*34ee2549SAlexey Bataev     // CHECK: [[CEIL_LOG2_10_INT:%.+]] = fptoui double [[CEIL_LOG2_10]] to i32
66*34ee2549SAlexey Bataev     // CHECK: br label %[[OUTER_BODY:[^,]+]]
67*34ee2549SAlexey Bataev     // CHECK: [[OUTER_BODY]]:
68*34ee2549SAlexey Bataev     // CHECK: [[K:%.+]] = phi i32 [ 0, %{{.+}} ], [ [[K_NEXT:%.+]], %{{.+}} ]
69*34ee2549SAlexey Bataev     // CHECK: [[K2POW:%.+]] = phi i64 [ 1, %{{.+}} ], [ [[K2POW_NEXT:%.+]], %{{.+}} ]
70*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp uge i64 9, [[K2POW]]
71*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[INNER_BODY:[^,]+]], label %[[INNER_EXIT:[^,]+]]
72*34ee2549SAlexey Bataev     // CHECK: [[INNER_BODY]]:
73*34ee2549SAlexey Bataev     // CHECK: [[I:%.+]] = phi i64 [ 9, %[[OUTER_BODY]] ], [ [[I_PREV:%.+]], %{{.+}} ]
74*34ee2549SAlexey Bataev 
75*34ee2549SAlexey Bataev     // a_buffer[i] += a_buffer[i-pow(2, k)];
76*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[I]], [[NUM_ELEMS]]
77*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
78*34ee2549SAlexey Bataev     // CHECK: [[IDX_SUB_K2POW:%.+]] = sub nuw i64 [[I]], [[K2POW]]
79*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[IDX_SUB_K2POW]], [[NUM_ELEMS]]
80*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
81*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[I]]
82*34ee2549SAlexey Bataev     // CHECK: [[IDX_SUB_K2POW:%.+]] = sub nuw i64 [[I]], [[K2POW]]
83*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_SUB_K2POW:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[IDX_SUB_K2POW]]
84*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_END:%.+]] = getelementptr float, float* [[A_BUF_IDX]], i64 [[NUM_ELEMS]]
85*34ee2549SAlexey Bataev     // CHECK: [[ISEMPTY:%.+]] = icmp eq float* [[A_BUF_IDX]], [[A_BUF_END]]
86*34ee2549SAlexey Bataev     // CHECK: br i1 [[ISEMPTY]], label %[[RED_DONE:[^,]+]], label %[[RED_BODY:[^,]+]]
87*34ee2549SAlexey Bataev     // CHECK: [[RED_BODY]]:
88*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW_ELEM:%.+]] = phi float* [ [[A_BUF_IDX_SUB_K2POW]], %[[INNER_BODY]] ], [ [[A_BUF_IDX_SUB_K2POW_NEXT:%.+]], %[[RED_BODY]] ]
89*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_ELEM:%.+]] = phi float* [ [[A_BUF_IDX]], %[[INNER_BODY]] ], [ [[A_BUF_IDX_NEXT:%.+]], %[[RED_BODY]] ]
90*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_VAL:%.+]] = load float, float* [[A_BUF_IDX_ELEM]],
91*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW_VAL:%.+]] = load float, float* [[A_BUF_IDX_SUB_K2POW_ELEM]],
92*34ee2549SAlexey Bataev     // CHECK: [[RED:%.+]] = fadd float [[A_BUF_IDX_VAL]], [[A_BUF_IDX_SUB_K2POW_VAL]]
93*34ee2549SAlexey Bataev     // CHECK: store float [[RED]], float* [[A_BUF_IDX_ELEM]],
94*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_NEXT]] = getelementptr float, float* [[A_BUF_IDX_ELEM]], i32 1
95*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW_NEXT]] = getelementptr float, float* [[A_BUF_IDX_SUB_K2POW_ELEM]], i32 1
96*34ee2549SAlexey Bataev     // CHECK: [[DONE:%.+]] = icmp eq float* [[A_BUF_IDX_NEXT]], [[A_BUF_END]]
97*34ee2549SAlexey Bataev     // CHECK: br i1 [[DONE]], label %[[RED_DONE]], label %[[RED_BODY]]
98*34ee2549SAlexey Bataev     // CHECK: [[RED_DONE]]:
99*34ee2549SAlexey Bataev 
100*34ee2549SAlexey Bataev     // b_buffer[i] += b_buffer[i-pow(2, k)];
101*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_VAL:%.+]] = load double, double* [[B_BUF_IDX]],
102*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_SUB_K2POW_VAL:%.+]] = load double, double* [[B_BUF_IDX_SUB_K2POW]],
103*34ee2549SAlexey Bataev     // CHECK: [[RED:%.+]] = fadd double [[B_BUF_IDX_VAL]], [[B_BUF_IDX_SUB_K2POW_VAL]]
104*34ee2549SAlexey Bataev     // CHECK: store double [[RED]], double* [[B_BUF_IDX]],
105*34ee2549SAlexey Bataev 
106*34ee2549SAlexey Bataev     // --i;
107*34ee2549SAlexey Bataev     // CHECK: [[I_PREV:%.+]] = sub nuw i64 [[I]], 1
108*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp uge i64 [[I_PREV]], [[K2POW]]
109*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[INNER_BODY]], label %[[INNER_EXIT]]
110*34ee2549SAlexey Bataev     // CHECK: [[INNER_EXIT]]:
111*34ee2549SAlexey Bataev 
112*34ee2549SAlexey Bataev     // ++k;
113*34ee2549SAlexey Bataev     // CHECK: [[K_NEXT]] = add nuw i32 [[K]], 1
114*34ee2549SAlexey Bataev     // k2pow <<= 1;
115*34ee2549SAlexey Bataev     // CHECK: [[K2POW_NEXT]] = shl nuw i64 [[K2POW]], 1
116*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp ne i32 [[K_NEXT]], [[CEIL_LOG2_10_INT]]
117*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[OUTER_BODY]], label %[[OUTER_EXIT:[^,]+]]
118*34ee2549SAlexey Bataev     // CHECK: [[OUTER_EXIT]]:
119*34ee2549SAlexey Bataev     bar();
120*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_init_4(
121*34ee2549SAlexey Bataev     // CHECK: call i8* @llvm.stacksave()
122*34ee2549SAlexey Bataev     // CHECK: store float 0.000000e+00, float* %
123*34ee2549SAlexey Bataev     // CHECK: store double 0.000000e+00, double* [[B_PRIV_ADDR:%.+]],
124*34ee2549SAlexey Bataev     // CHECK: br label %[[DISPATCH:[^,]+]]
125*34ee2549SAlexey Bataev 
126*34ee2549SAlexey Bataev     // Skip the before scan body.
127*34ee2549SAlexey Bataev     // CHECK: call void @{{.+}}foo{{.+}}()
128*34ee2549SAlexey Bataev 
129*34ee2549SAlexey Bataev     // CHECK: [[EXIT_INSCAN:[^,]+]]:
130*34ee2549SAlexey Bataev     // CHECK: br label %[[LOOP_CONTINUE:[^,]+]]
131*34ee2549SAlexey Bataev 
132*34ee2549SAlexey Bataev     // CHECK: [[DISPATCH]]:
133*34ee2549SAlexey Bataev     // a_priv[[0..n] = a_buffer[i][0..n];
134*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX_I:%.+]] = load i32, i32* [[IV_ADDR:%.+]],
135*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX:%.+]] = zext i32 [[BASE_IDX_I]] to i64
136*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[BASE_IDX]], [[NUM_ELEMS]]
137*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
138*34ee2549SAlexey Bataev     // CHECK: [[A_PRIV:%.+]] = getelementptr inbounds [10 x float], [10 x float]* [[A_PRIV_ADDR:%.+]], i64 0, i64 0
139*34ee2549SAlexey Bataev     // CHECK: [[BYTES:%.+]] = mul nuw i64 [[NUM_ELEMS:%.+]], 4
140*34ee2549SAlexey Bataev     // CHECK: [[DEST:%.+]] = bitcast float* [[A_PRIV]] to i8*
141*34ee2549SAlexey Bataev     // CHECK: [[SRC:%.+]] = bitcast float* [[A_BUF_IDX]] to i8*
142*34ee2549SAlexey Bataev     // CHECK: call void @llvm.memcpy.p0i8.p0i8.i64(i8* {{.*}}[[DEST]], i8* {{.*}}[[SRC]], i64 [[BYTES]], i1 false)
143*34ee2549SAlexey Bataev 
144*34ee2549SAlexey Bataev     // b_priv = b_buffer[i];
145*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[BASE_IDX]]
146*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_VAL:%.+]] = load double, double* [[B_BUF_IDX]],
147*34ee2549SAlexey Bataev     // CHECK: store double [[B_BUF_IDX_VAL]], double* [[B_PRIV_ADDR]],
148*34ee2549SAlexey Bataev     // CHECK: br label %[[SCAN_PHASE:[^,]+]]
149*34ee2549SAlexey Bataev 
150*34ee2549SAlexey Bataev     // CHECK: [[SCAN_PHASE]]:
151*34ee2549SAlexey Bataev     // CHECK: call void @{{.+}}bar{{.+}}()
152*34ee2549SAlexey Bataev     // CHECK: br label %[[EXIT_INSCAN]]
153*34ee2549SAlexey Bataev 
154*34ee2549SAlexey Bataev     // CHECK: [[LOOP_CONTINUE]]:
155*34ee2549SAlexey Bataev     // CHECK: call void @llvm.stackrestore(i8* %
156*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_fini(
157*34ee2549SAlexey Bataev     // CHECK: call void @llvm.stackrestore(i8*
158*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_barrier(
159*34ee2549SAlexey Bataev   }
160*34ee2549SAlexey Bataev 
161*34ee2549SAlexey Bataev   // CHECK: call i8* @llvm.stacksave()
162*34ee2549SAlexey Bataev   // CHECK: [[A_BUF_SIZE:%.+]] = mul nuw i64 10, [[NUM_ELEMS:%[^,]+]]
163*34ee2549SAlexey Bataev 
164*34ee2549SAlexey Bataev   // float a_buffer[10][n];
165*34ee2549SAlexey Bataev   // CHECK: [[A_BUF:%.+]] = alloca float, i64 [[A_BUF_SIZE]],
166*34ee2549SAlexey Bataev 
167*34ee2549SAlexey Bataev   // double b_buffer[10];
168*34ee2549SAlexey Bataev   // CHECK: [[B_BUF:%.+]] = alloca double, i64 10,
169*34ee2549SAlexey Bataev #pragma omp for simd reduction(inscan, +:a[:n], b)
170*34ee2549SAlexey Bataev   for (int i = 0; i < 10; ++i) {
171*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_init_4(
172*34ee2549SAlexey Bataev     // CHECK: call i8* @llvm.stacksave()
173*34ee2549SAlexey Bataev     // CHECK: store float 0.000000e+00, float* %
174*34ee2549SAlexey Bataev     // CHECK: store double 0.000000e+00, double* [[B_PRIV_ADDR:%.+]],
175*34ee2549SAlexey Bataev     // CHECK: br label %[[DISPATCH:[^,]+]]
176*34ee2549SAlexey Bataev 
177*34ee2549SAlexey Bataev     // Skip the before scan body.
178*34ee2549SAlexey Bataev     // CHECK: call void @{{.+}}foo{{.+}}()
179*34ee2549SAlexey Bataev 
180*34ee2549SAlexey Bataev     // CHECK: [[EXIT_INSCAN:[^,]+]]:
181*34ee2549SAlexey Bataev 
182*34ee2549SAlexey Bataev     // a_buffer[i][0..n] = a_priv[[0..n];
183*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX_I:%.+]] = load i32, i32* [[IV_ADDR:%.+]],
184*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX:%.+]] = zext i32 [[BASE_IDX_I]] to i64
185*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[BASE_IDX]], [[NUM_ELEMS]]
186*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
187*34ee2549SAlexey Bataev     // CHECK: [[A_PRIV:%.+]] = getelementptr inbounds [10 x float], [10 x float]* [[A_PRIV_ADDR:%.+]], i64 0, i64 0
188*34ee2549SAlexey Bataev     // CHECK: [[BYTES:%.+]] = mul nuw i64 [[NUM_ELEMS:%.+]], 4
189*34ee2549SAlexey Bataev     // CHECK: [[DEST:%.+]] = bitcast float* [[A_BUF_IDX]] to i8*
190*34ee2549SAlexey Bataev     // CHECK: [[SRC:%.+]] = bitcast float* [[A_PRIV]] to i8*
191*34ee2549SAlexey Bataev     // CHECK: call void @llvm.memcpy.p0i8.p0i8.i64(i8* {{.*}}[[DEST]], i8* {{.*}}[[SRC]], i64 [[BYTES]], i1 false)
192*34ee2549SAlexey Bataev 
193*34ee2549SAlexey Bataev     // b_buffer[i] = b_priv;
194*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[BASE_IDX]]
195*34ee2549SAlexey Bataev     // CHECK: [[B_PRIV:%.+]] = load double, double* [[B_PRIV_ADDR]],
196*34ee2549SAlexey Bataev     // CHECK: store double [[B_PRIV]], double* [[B_BUF_IDX]],
197*34ee2549SAlexey Bataev     // CHECK: br label %[[LOOP_CONTINUE:[^,]+]]
198*34ee2549SAlexey Bataev 
199*34ee2549SAlexey Bataev     // CHECK: [[DISPATCH]]:
200*34ee2549SAlexey Bataev     // CHECK: br label %[[INPUT_PHASE:[^,]+]]
201*34ee2549SAlexey Bataev 
202*34ee2549SAlexey Bataev     // CHECK: [[INPUT_PHASE]]:
203*34ee2549SAlexey Bataev     // CHECK: call void @{{.+}}bar{{.+}}()
204*34ee2549SAlexey Bataev     // CHECK: br label %[[EXIT_INSCAN]]
205*34ee2549SAlexey Bataev 
206*34ee2549SAlexey Bataev     // CHECK: [[LOOP_CONTINUE]]:
207*34ee2549SAlexey Bataev     // CHECK: call void @llvm.stackrestore(i8* %
208*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_fini(
209*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_barrier(
210*34ee2549SAlexey Bataev     foo();
211*34ee2549SAlexey Bataev #pragma omp scan exclusive(a[:n], b)
212*34ee2549SAlexey Bataev     // CHECK: [[LOG2_10:%.+]] = call double @llvm.log2.f64(double 1.000000e+01)
213*34ee2549SAlexey Bataev     // CHECK: [[CEIL_LOG2_10:%.+]] = call double @llvm.ceil.f64(double [[LOG2_10]])
214*34ee2549SAlexey Bataev     // CHECK: [[CEIL_LOG2_10_INT:%.+]] = fptoui double [[CEIL_LOG2_10]] to i32
215*34ee2549SAlexey Bataev     // CHECK: br label %[[OUTER_BODY:[^,]+]]
216*34ee2549SAlexey Bataev     // CHECK: [[OUTER_BODY]]:
217*34ee2549SAlexey Bataev     // CHECK: [[K:%.+]] = phi i32 [ 0, %{{.+}} ], [ [[K_NEXT:%.+]], %{{.+}} ]
218*34ee2549SAlexey Bataev     // CHECK: [[K2POW:%.+]] = phi i64 [ 1, %{{.+}} ], [ [[K2POW_NEXT:%.+]], %{{.+}} ]
219*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp uge i64 9, [[K2POW]]
220*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[INNER_BODY:[^,]+]], label %[[INNER_EXIT:[^,]+]]
221*34ee2549SAlexey Bataev     // CHECK: [[INNER_BODY]]:
222*34ee2549SAlexey Bataev     // CHECK: [[I:%.+]] = phi i64 [ 9, %[[OUTER_BODY]] ], [ [[I_PREV:%.+]], %{{.+}} ]
223*34ee2549SAlexey Bataev 
224*34ee2549SAlexey Bataev     // a_buffer[i] += a_buffer[i-pow(2, k)];
225*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[I]], [[NUM_ELEMS]]
226*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
227*34ee2549SAlexey Bataev     // CHECK: [[IDX_SUB_K2POW:%.+]] = sub nuw i64 [[I]], [[K2POW]]
228*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[IDX_SUB_K2POW]], [[NUM_ELEMS]]
229*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
230*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[I]]
231*34ee2549SAlexey Bataev     // CHECK: [[IDX_SUB_K2POW:%.+]] = sub nuw i64 [[I]], [[K2POW]]
232*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_SUB_K2POW:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[IDX_SUB_K2POW]]
233*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_END:%.+]] = getelementptr float, float* [[A_BUF_IDX]], i64 [[NUM_ELEMS]]
234*34ee2549SAlexey Bataev     // CHECK: [[ISEMPTY:%.+]] = icmp eq float* [[A_BUF_IDX]], [[A_BUF_END]]
235*34ee2549SAlexey Bataev     // CHECK: br i1 [[ISEMPTY]], label %[[RED_DONE:[^,]+]], label %[[RED_BODY:[^,]+]]
236*34ee2549SAlexey Bataev     // CHECK: [[RED_BODY]]:
237*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW_ELEM:%.+]] = phi float* [ [[A_BUF_IDX_SUB_K2POW]], %[[INNER_BODY]] ], [ [[A_BUF_IDX_SUB_K2POW_NEXT:%.+]], %[[RED_BODY]] ]
238*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_ELEM:%.+]] = phi float* [ [[A_BUF_IDX]], %[[INNER_BODY]] ], [ [[A_BUF_IDX_NEXT:%.+]], %[[RED_BODY]] ]
239*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_VAL:%.+]] = load float, float* [[A_BUF_IDX_ELEM]],
240*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW_VAL:%.+]] = load float, float* [[A_BUF_IDX_SUB_K2POW_ELEM]],
241*34ee2549SAlexey Bataev     // CHECK: [[RED:%.+]] = fadd float [[A_BUF_IDX_VAL]], [[A_BUF_IDX_SUB_K2POW_VAL]]
242*34ee2549SAlexey Bataev     // CHECK: store float [[RED]], float* [[A_BUF_IDX_ELEM]],
243*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_NEXT]] = getelementptr float, float* [[A_BUF_IDX_ELEM]], i32 1
244*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX_SUB_K2POW_NEXT]] = getelementptr float, float* [[A_BUF_IDX_SUB_K2POW_ELEM]], i32 1
245*34ee2549SAlexey Bataev     // CHECK: [[DONE:%.+]] = icmp eq float* [[A_BUF_IDX_NEXT]], [[A_BUF_END]]
246*34ee2549SAlexey Bataev     // CHECK: br i1 [[DONE]], label %[[RED_DONE]], label %[[RED_BODY]]
247*34ee2549SAlexey Bataev     // CHECK: [[RED_DONE]]:
248*34ee2549SAlexey Bataev 
249*34ee2549SAlexey Bataev     // b_buffer[i] += b_buffer[i-pow(2, k)];
250*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_VAL:%.+]] = load double, double* [[B_BUF_IDX]],
251*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_SUB_K2POW_VAL:%.+]] = load double, double* [[B_BUF_IDX_SUB_K2POW]],
252*34ee2549SAlexey Bataev     // CHECK: [[RED:%.+]] = fadd double [[B_BUF_IDX_VAL]], [[B_BUF_IDX_SUB_K2POW_VAL]]
253*34ee2549SAlexey Bataev     // CHECK: store double [[RED]], double* [[B_BUF_IDX]],
254*34ee2549SAlexey Bataev 
255*34ee2549SAlexey Bataev     // --i;
256*34ee2549SAlexey Bataev     // CHECK: [[I_PREV:%.+]] = sub nuw i64 [[I]], 1
257*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp uge i64 [[I_PREV]], [[K2POW]]
258*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[INNER_BODY]], label %[[INNER_EXIT]]
259*34ee2549SAlexey Bataev     // CHECK: [[INNER_EXIT]]:
260*34ee2549SAlexey Bataev 
261*34ee2549SAlexey Bataev     // ++k;
262*34ee2549SAlexey Bataev     // CHECK: [[K_NEXT]] = add nuw i32 [[K]], 1
263*34ee2549SAlexey Bataev     // k2pow <<= 1;
264*34ee2549SAlexey Bataev     // CHECK: [[K2POW_NEXT]] = shl nuw i64 [[K2POW]], 1
265*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp ne i32 [[K_NEXT]], [[CEIL_LOG2_10_INT]]
266*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[OUTER_BODY]], label %[[OUTER_EXIT:[^,]+]]
267*34ee2549SAlexey Bataev     // CHECK: [[OUTER_EXIT]]:
268*34ee2549SAlexey Bataev     bar();
269*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_init_4(
270*34ee2549SAlexey Bataev     // CHECK: call i8* @llvm.stacksave()
271*34ee2549SAlexey Bataev     // CHECK: store float 0.000000e+00, float* %
272*34ee2549SAlexey Bataev     // CHECK: store double 0.000000e+00, double* [[B_PRIV_ADDR:%.+]],
273*34ee2549SAlexey Bataev     // CHECK: br label %[[DISPATCH:[^,]+]]
274*34ee2549SAlexey Bataev 
275*34ee2549SAlexey Bataev     // CHECK: [[SCAN_PHASE:.+]]:
276*34ee2549SAlexey Bataev     // CHECK: call void @{{.+}}foo{{.+}}()
277*34ee2549SAlexey Bataev     // CHECK: br label %[[LOOP_CONTINUE:.+]]
278*34ee2549SAlexey Bataev 
279*34ee2549SAlexey Bataev     // CHECK: [[DISPATCH]]:
280*34ee2549SAlexey Bataev     // if (i >0)
281*34ee2549SAlexey Bataev     //   a_priv[[0..n] = a_buffer[i-1][0..n];
282*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX_I:%.+]] = load i32, i32* [[IV_ADDR:%.+]],
283*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX:%.+]] = zext i32 [[BASE_IDX_I]] to i64
284*34ee2549SAlexey Bataev     // CHECK: [[CMP:%.+]] = icmp eq i64 [[BASE_IDX]], 0
285*34ee2549SAlexey Bataev     // CHECK: br i1 [[CMP]], label %[[IF_DONE:[^,]+]], label %[[IF_THEN:[^,]+]]
286*34ee2549SAlexey Bataev     // CHECK: [[IF_THEN]]:
287*34ee2549SAlexey Bataev     // CHECK: [[BASE_IDX_SUB_1:%.+]] = sub nuw i64 [[BASE_IDX]], 1
288*34ee2549SAlexey Bataev     // CHECK: [[IDX:%.+]] = mul nsw i64 [[BASE_IDX_SUB_1]], [[NUM_ELEMS]]
289*34ee2549SAlexey Bataev     // CHECK: [[A_BUF_IDX:%.+]] = getelementptr inbounds float, float* [[A_BUF]], i64 [[IDX]]
290*34ee2549SAlexey Bataev     // CHECK: [[A_PRIV:%.+]] = getelementptr inbounds [10 x float], [10 x float]* [[A_PRIV_ADDR:%.+]], i64 0, i64 0
291*34ee2549SAlexey Bataev     // CHECK: [[BYTES:%.+]] = mul nuw i64 [[NUM_ELEMS:%.+]], 4
292*34ee2549SAlexey Bataev     // CHECK: [[DEST:%.+]] = bitcast float* [[A_PRIV]] to i8*
293*34ee2549SAlexey Bataev     // CHECK: [[SRC:%.+]] = bitcast float* [[A_BUF_IDX]] to i8*
294*34ee2549SAlexey Bataev     // CHECK: call void @llvm.memcpy.p0i8.p0i8.i64(i8* {{.*}}[[DEST]], i8* {{.*}}[[SRC]], i64 [[BYTES]], i1 false)
295*34ee2549SAlexey Bataev 
296*34ee2549SAlexey Bataev     // b_priv = b_buffer[i];
297*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX:%.+]] = getelementptr inbounds double, double* [[B_BUF]], i64 [[BASE_IDX_SUB_1]]
298*34ee2549SAlexey Bataev     // CHECK: [[B_BUF_IDX_VAL:%.+]] = load double, double* [[B_BUF_IDX]],
299*34ee2549SAlexey Bataev     // CHECK: store double [[B_BUF_IDX_VAL]], double* [[B_PRIV_ADDR]],
300*34ee2549SAlexey Bataev     // CHECK: br label %[[SCAN_PHASE]]
301*34ee2549SAlexey Bataev 
302*34ee2549SAlexey Bataev     // CHECK: [[LOOP_CONTINUE]]:
303*34ee2549SAlexey Bataev     // CHECK: call void @llvm.stackrestore(i8* %
304*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_for_static_fini(
305*34ee2549SAlexey Bataev     // CHECK: call void @llvm.stackrestore(i8*
306*34ee2549SAlexey Bataev     // CHECK: call void @__kmpc_barrier(
307*34ee2549SAlexey Bataev   }
308*34ee2549SAlexey Bataev }
309*34ee2549SAlexey Bataev 
310*34ee2549SAlexey Bataev #endif
311*34ee2549SAlexey Bataev // CHECK: !{!"llvm.loop.vectorize.enable", i1 true}
312*34ee2549SAlexey Bataev 
313