1 // NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py UTC_ARGS: --function-signature --include-generated-funcs
2 // RUN: %clang_cc1 -fopenmp-enable-irbuilder -verify -fopenmp -fopenmp-version=50 -x c++ -triple x86_64-unknown-unknown -emit-llvm %s -o - | FileCheck %s
3 // expected-no-diagnostics
4 
5 #ifndef HEADER
6 #define HEADER
7 
8 struct MyIterator {
9   MyIterator(unsigned pos);
10   MyIterator(const MyIterator &other);
11   const MyIterator &operator=(const MyIterator &that);
12   MyIterator &operator++();
13   int operator-(const MyIterator &that) const;
14   MyIterator &operator+=(unsigned a);
15   MyIterator operator+(unsigned a) const;
16   bool operator==(const MyIterator &that) const;
17   bool operator!=(const MyIterator &that) const;
18   unsigned operator*() const;
19 };
20 
21 struct MyRange {
22   MyRange(int n);
23 
24   MyIterator begin();
25   MyIterator end();
26 };
27 
workshareloop_rangefor(float * a,float * b,float * c)28 extern "C" void workshareloop_rangefor(float *a, float *b, float *c) {
29 #pragma omp for
30   for (unsigned i : MyRange(42)) {
31     a[i] = b[i] * c[i];
32   }
33 }
34 
35 #endif // HEADER
36 // CHECK-LABEL: define {{[^@]+}}@workshareloop_rangefor
37 // CHECK-SAME: (float* [[A:%.*]], float* [[B:%.*]], float* [[C:%.*]]) #[[ATTR0:[0-9]+]] {
38 // CHECK-NEXT:  entry:
39 // CHECK-NEXT:    [[A_ADDR:%.*]] = alloca float*, align 8
40 // CHECK-NEXT:    [[B_ADDR:%.*]] = alloca float*, align 8
41 // CHECK-NEXT:    [[C_ADDR:%.*]] = alloca float*, align 8
42 // CHECK-NEXT:    [[__RANGE2:%.*]] = alloca %struct.MyRange*, align 8
43 // CHECK-NEXT:    [[REF_TMP:%.*]] = alloca [[STRUCT_MYRANGE:%.*]], align 1
44 // CHECK-NEXT:    [[__BEGIN2:%.*]] = alloca [[STRUCT_MYITERATOR:%.*]], align 1
45 // CHECK-NEXT:    [[__END2:%.*]] = alloca [[STRUCT_MYITERATOR]], align 1
46 // CHECK-NEXT:    [[I:%.*]] = alloca i32, align 4
47 // CHECK-NEXT:    [[AGG_CAPTURED:%.*]] = alloca [[STRUCT_ANON:%.*]], align 8
48 // CHECK-NEXT:    [[AGG_CAPTURED1:%.*]] = alloca [[STRUCT_ANON_0:%.*]], align 1
49 // CHECK-NEXT:    [[DOTCOUNT_ADDR:%.*]] = alloca i64, align 8
50 // CHECK-NEXT:    [[P_LASTITER:%.*]] = alloca i32, align 4
51 // CHECK-NEXT:    [[P_LOWERBOUND:%.*]] = alloca i64, align 8
52 // CHECK-NEXT:    [[P_UPPERBOUND:%.*]] = alloca i64, align 8
53 // CHECK-NEXT:    [[P_STRIDE:%.*]] = alloca i64, align 8
54 // CHECK-NEXT:    store float* [[A]], float** [[A_ADDR]], align 8
55 // CHECK-NEXT:    store float* [[B]], float** [[B_ADDR]], align 8
56 // CHECK-NEXT:    store float* [[C]], float** [[C_ADDR]], align 8
57 // CHECK-NEXT:    call void @_ZN7MyRangeC1Ei(%struct.MyRange* nonnull align 1 dereferenceable(1) [[REF_TMP]], i32 42)
58 // CHECK-NEXT:    store %struct.MyRange* [[REF_TMP]], %struct.MyRange** [[__RANGE2]], align 8
59 // CHECK-NEXT:    [[TMP0:%.*]] = load %struct.MyRange*, %struct.MyRange** [[__RANGE2]], align 8
60 // CHECK-NEXT:    call void @_ZN7MyRange5beginEv(%struct.MyIterator* sret([[STRUCT_MYITERATOR]]) align 1 [[__BEGIN2]], %struct.MyRange* nonnull align 1 dereferenceable(1) [[TMP0]])
61 // CHECK-NEXT:    [[TMP1:%.*]] = load %struct.MyRange*, %struct.MyRange** [[__RANGE2]], align 8
62 // CHECK-NEXT:    call void @_ZN7MyRange3endEv(%struct.MyIterator* sret([[STRUCT_MYITERATOR]]) align 1 [[__END2]], %struct.MyRange* nonnull align 1 dereferenceable(1) [[TMP1]])
63 // CHECK-NEXT:    [[CALL:%.*]] = call i32 @_ZNK10MyIteratordeEv(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[__BEGIN2]])
64 // CHECK-NEXT:    store i32 [[CALL]], i32* [[I]], align 4
65 // CHECK-NEXT:    [[TMP2:%.*]] = getelementptr inbounds [[STRUCT_ANON]], %struct.anon* [[AGG_CAPTURED]], i32 0, i32 0
66 // CHECK-NEXT:    store %struct.MyIterator* [[__BEGIN2]], %struct.MyIterator** [[TMP2]], align 8
67 // CHECK-NEXT:    [[TMP3:%.*]] = getelementptr inbounds [[STRUCT_ANON]], %struct.anon* [[AGG_CAPTURED]], i32 0, i32 1
68 // CHECK-NEXT:    store %struct.MyIterator* [[__END2]], %struct.MyIterator** [[TMP3]], align 8
69 // CHECK-NEXT:    [[TMP4:%.*]] = getelementptr inbounds [[STRUCT_ANON_0]], %struct.anon.0* [[AGG_CAPTURED1]], i32 0, i32 0
70 // CHECK-NEXT:    call void @_ZN10MyIteratorC1ERKS_(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[TMP4]], %struct.MyIterator* nonnull align 1 dereferenceable(1) [[__BEGIN2]])
71 // CHECK-NEXT:    call void @__captured_stmt(i64* [[DOTCOUNT_ADDR]], %struct.anon* [[AGG_CAPTURED]])
72 // CHECK-NEXT:    [[DOTCOUNT:%.*]] = load i64, i64* [[DOTCOUNT_ADDR]], align 8
73 // CHECK-NEXT:    br label [[OMP_LOOP_PREHEADER:%.*]]
74 // CHECK:       omp_loop.preheader:
75 // CHECK-NEXT:    store i64 0, i64* [[P_LOWERBOUND]], align 8
76 // CHECK-NEXT:    [[TMP5:%.*]] = sub i64 [[DOTCOUNT]], 1
77 // CHECK-NEXT:    store i64 [[TMP5]], i64* [[P_UPPERBOUND]], align 8
78 // CHECK-NEXT:    store i64 1, i64* [[P_STRIDE]], align 8
79 // CHECK-NEXT:    [[OMP_GLOBAL_THREAD_NUM:%.*]] = call i32 @__kmpc_global_thread_num(%struct.ident_t* @[[GLOB1:[0-9]+]])
80 // CHECK-NEXT:    call void @__kmpc_for_static_init_8u(%struct.ident_t* @[[GLOB1]], i32 [[OMP_GLOBAL_THREAD_NUM]], i32 34, i32* [[P_LASTITER]], i64* [[P_LOWERBOUND]], i64* [[P_UPPERBOUND]], i64* [[P_STRIDE]], i64 1, i64 1)
81 // CHECK-NEXT:    [[TMP6:%.*]] = load i64, i64* [[P_LOWERBOUND]], align 8
82 // CHECK-NEXT:    [[TMP7:%.*]] = load i64, i64* [[P_UPPERBOUND]], align 8
83 // CHECK-NEXT:    [[TMP8:%.*]] = sub i64 [[TMP7]], [[TMP6]]
84 // CHECK-NEXT:    [[TMP9:%.*]] = add i64 [[TMP8]], 1
85 // CHECK-NEXT:    br label [[OMP_LOOP_HEADER:%.*]]
86 // CHECK:       omp_loop.header:
87 // CHECK-NEXT:    [[OMP_LOOP_IV:%.*]] = phi i64 [ 0, [[OMP_LOOP_PREHEADER]] ], [ [[OMP_LOOP_NEXT:%.*]], [[OMP_LOOP_INC:%.*]] ]
88 // CHECK-NEXT:    br label [[OMP_LOOP_COND:%.*]]
89 // CHECK:       omp_loop.cond:
90 // CHECK-NEXT:    [[OMP_LOOP_CMP:%.*]] = icmp ult i64 [[OMP_LOOP_IV]], [[TMP9]]
91 // CHECK-NEXT:    br i1 [[OMP_LOOP_CMP]], label [[OMP_LOOP_BODY:%.*]], label [[OMP_LOOP_EXIT:%.*]]
92 // CHECK:       omp_loop.body:
93 // CHECK-NEXT:    [[TMP10:%.*]] = add i64 [[OMP_LOOP_IV]], [[TMP6]]
94 // CHECK-NEXT:    call void @__captured_stmt.1(i32* [[I]], i64 [[TMP10]], %struct.anon.0* [[AGG_CAPTURED1]])
95 // CHECK-NEXT:    [[TMP11:%.*]] = load float*, float** [[B_ADDR]], align 8
96 // CHECK-NEXT:    [[TMP12:%.*]] = load i32, i32* [[I]], align 4
97 // CHECK-NEXT:    [[IDXPROM:%.*]] = zext i32 [[TMP12]] to i64
98 // CHECK-NEXT:    [[ARRAYIDX:%.*]] = getelementptr inbounds float, float* [[TMP11]], i64 [[IDXPROM]]
99 // CHECK-NEXT:    [[TMP13:%.*]] = load float, float* [[ARRAYIDX]], align 4
100 // CHECK-NEXT:    [[TMP14:%.*]] = load float*, float** [[C_ADDR]], align 8
101 // CHECK-NEXT:    [[TMP15:%.*]] = load i32, i32* [[I]], align 4
102 // CHECK-NEXT:    [[IDXPROM2:%.*]] = zext i32 [[TMP15]] to i64
103 // CHECK-NEXT:    [[ARRAYIDX3:%.*]] = getelementptr inbounds float, float* [[TMP14]], i64 [[IDXPROM2]]
104 // CHECK-NEXT:    [[TMP16:%.*]] = load float, float* [[ARRAYIDX3]], align 4
105 // CHECK-NEXT:    [[MUL:%.*]] = fmul float [[TMP13]], [[TMP16]]
106 // CHECK-NEXT:    [[TMP17:%.*]] = load float*, float** [[A_ADDR]], align 8
107 // CHECK-NEXT:    [[TMP18:%.*]] = load i32, i32* [[I]], align 4
108 // CHECK-NEXT:    [[IDXPROM4:%.*]] = zext i32 [[TMP18]] to i64
109 // CHECK-NEXT:    [[ARRAYIDX5:%.*]] = getelementptr inbounds float, float* [[TMP17]], i64 [[IDXPROM4]]
110 // CHECK-NEXT:    store float [[MUL]], float* [[ARRAYIDX5]], align 4
111 // CHECK-NEXT:    br label [[OMP_LOOP_INC]]
112 // CHECK:       omp_loop.inc:
113 // CHECK-NEXT:    [[OMP_LOOP_NEXT]] = add nuw i64 [[OMP_LOOP_IV]], 1
114 // CHECK-NEXT:    br label [[OMP_LOOP_HEADER]]
115 // CHECK:       omp_loop.exit:
116 // CHECK-NEXT:    call void @__kmpc_for_static_fini(%struct.ident_t* @[[GLOB1]], i32 [[OMP_GLOBAL_THREAD_NUM]])
117 // CHECK-NEXT:    [[OMP_GLOBAL_THREAD_NUM6:%.*]] = call i32 @__kmpc_global_thread_num(%struct.ident_t* @[[GLOB1]])
118 // CHECK-NEXT:    call void @__kmpc_barrier(%struct.ident_t* @[[GLOB2:[0-9]+]], i32 [[OMP_GLOBAL_THREAD_NUM6]])
119 // CHECK-NEXT:    br label [[OMP_LOOP_AFTER:%.*]]
120 // CHECK:       omp_loop.after:
121 // CHECK-NEXT:    ret void
122 //
123 //
124 // CHECK-LABEL: define {{[^@]+}}@__captured_stmt
125 // CHECK-SAME: (i64* nonnull align 8 dereferenceable(8) [[DISTANCE:%.*]], %struct.anon* noalias [[__CONTEXT:%.*]]) #[[ATTR2:[0-9]+]] {
126 // CHECK-NEXT:  entry:
127 // CHECK-NEXT:    [[DISTANCE_ADDR:%.*]] = alloca i64*, align 8
128 // CHECK-NEXT:    [[__CONTEXT_ADDR:%.*]] = alloca %struct.anon*, align 8
129 // CHECK-NEXT:    [[DOTSTART:%.*]] = alloca [[STRUCT_MYITERATOR:%.*]], align 1
130 // CHECK-NEXT:    [[DOTSTOP:%.*]] = alloca [[STRUCT_MYITERATOR]], align 1
131 // CHECK-NEXT:    [[DOTSTEP:%.*]] = alloca i64, align 8
132 // CHECK-NEXT:    store i64* [[DISTANCE]], i64** [[DISTANCE_ADDR]], align 8
133 // CHECK-NEXT:    store %struct.anon* [[__CONTEXT]], %struct.anon** [[__CONTEXT_ADDR]], align 8
134 // CHECK-NEXT:    [[TMP0:%.*]] = load %struct.anon*, %struct.anon** [[__CONTEXT_ADDR]], align 8
135 // CHECK-NEXT:    [[TMP1:%.*]] = getelementptr inbounds [[STRUCT_ANON:%.*]], %struct.anon* [[TMP0]], i32 0, i32 0
136 // CHECK-NEXT:    [[TMP2:%.*]] = load %struct.MyIterator*, %struct.MyIterator** [[TMP1]], align 8
137 // CHECK-NEXT:    call void @_ZN10MyIteratorC1ERKS_(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[DOTSTART]], %struct.MyIterator* nonnull align 1 dereferenceable(1) [[TMP2]])
138 // CHECK-NEXT:    [[TMP3:%.*]] = getelementptr inbounds [[STRUCT_ANON]], %struct.anon* [[TMP0]], i32 0, i32 1
139 // CHECK-NEXT:    [[TMP4:%.*]] = load %struct.MyIterator*, %struct.MyIterator** [[TMP3]], align 8
140 // CHECK-NEXT:    call void @_ZN10MyIteratorC1ERKS_(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[DOTSTOP]], %struct.MyIterator* nonnull align 1 dereferenceable(1) [[TMP4]])
141 // CHECK-NEXT:    store i64 1, i64* [[DOTSTEP]], align 8
142 // CHECK-NEXT:    [[TMP5:%.*]] = load i64, i64* [[DOTSTEP]], align 8
143 // CHECK-NEXT:    [[CMP:%.*]] = icmp ult i64 [[TMP5]], 0
144 // CHECK-NEXT:    br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_FALSE:%.*]]
145 // CHECK:       cond.true:
146 // CHECK-NEXT:    [[CALL:%.*]] = call i32 @_ZNK10MyIteratormiERKS_(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[DOTSTART]], %struct.MyIterator* nonnull align 1 dereferenceable(1) [[DOTSTOP]])
147 // CHECK-NEXT:    [[CONV:%.*]] = sext i32 [[CALL]] to i64
148 // CHECK-NEXT:    [[TMP6:%.*]] = load i64, i64* [[DOTSTEP]], align 8
149 // CHECK-NEXT:    [[SUB:%.*]] = sub i64 0, [[TMP6]]
150 // CHECK-NEXT:    [[DIV:%.*]] = udiv i64 [[CONV]], [[SUB]]
151 // CHECK-NEXT:    br label [[COND_END:%.*]]
152 // CHECK:       cond.false:
153 // CHECK-NEXT:    [[CALL1:%.*]] = call i32 @_ZNK10MyIteratormiERKS_(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[DOTSTOP]], %struct.MyIterator* nonnull align 1 dereferenceable(1) [[DOTSTART]])
154 // CHECK-NEXT:    [[CONV2:%.*]] = sext i32 [[CALL1]] to i64
155 // CHECK-NEXT:    [[TMP7:%.*]] = load i64, i64* [[DOTSTEP]], align 8
156 // CHECK-NEXT:    [[DIV3:%.*]] = udiv i64 [[CONV2]], [[TMP7]]
157 // CHECK-NEXT:    br label [[COND_END]]
158 // CHECK:       cond.end:
159 // CHECK-NEXT:    [[COND:%.*]] = phi i64 [ [[DIV]], [[COND_TRUE]] ], [ [[DIV3]], [[COND_FALSE]] ]
160 // CHECK-NEXT:    [[TMP8:%.*]] = load i64*, i64** [[DISTANCE_ADDR]], align 8
161 // CHECK-NEXT:    store i64 [[COND]], i64* [[TMP8]], align 8
162 // CHECK-NEXT:    ret void
163 //
164 //
165 // CHECK-LABEL: define {{[^@]+}}@__captured_stmt.1
166 // CHECK-SAME: (i32* nonnull align 4 dereferenceable(4) [[LOOPVAR:%.*]], i64 [[LOGICAL:%.*]], %struct.anon.0* noalias [[__CONTEXT:%.*]]) #[[ATTR2]] {
167 // CHECK-NEXT:  entry:
168 // CHECK-NEXT:    [[LOOPVAR_ADDR:%.*]] = alloca i32*, align 8
169 // CHECK-NEXT:    [[LOGICAL_ADDR:%.*]] = alloca i64, align 8
170 // CHECK-NEXT:    [[__CONTEXT_ADDR:%.*]] = alloca %struct.anon.0*, align 8
171 // CHECK-NEXT:    [[REF_TMP:%.*]] = alloca [[STRUCT_MYITERATOR:%.*]], align 1
172 // CHECK-NEXT:    store i32* [[LOOPVAR]], i32** [[LOOPVAR_ADDR]], align 8
173 // CHECK-NEXT:    store i64 [[LOGICAL]], i64* [[LOGICAL_ADDR]], align 8
174 // CHECK-NEXT:    store %struct.anon.0* [[__CONTEXT]], %struct.anon.0** [[__CONTEXT_ADDR]], align 8
175 // CHECK-NEXT:    [[TMP0:%.*]] = load %struct.anon.0*, %struct.anon.0** [[__CONTEXT_ADDR]], align 8
176 // CHECK-NEXT:    [[TMP1:%.*]] = getelementptr inbounds [[STRUCT_ANON_0:%.*]], %struct.anon.0* [[TMP0]], i32 0, i32 0
177 // CHECK-NEXT:    [[TMP2:%.*]] = load i64, i64* [[LOGICAL_ADDR]], align 8
178 // CHECK-NEXT:    [[MUL:%.*]] = mul i64 1, [[TMP2]]
179 // CHECK-NEXT:    [[CONV:%.*]] = trunc i64 [[MUL]] to i32
180 // CHECK-NEXT:    call void @_ZNK10MyIteratorplEj(%struct.MyIterator* sret([[STRUCT_MYITERATOR]]) align 1 [[REF_TMP]], %struct.MyIterator* nonnull align 1 dereferenceable(1) [[TMP1]], i32 [[CONV]])
181 // CHECK-NEXT:    [[CALL:%.*]] = call i32 @_ZNK10MyIteratordeEv(%struct.MyIterator* nonnull align 1 dereferenceable(1) [[REF_TMP]])
182 // CHECK-NEXT:    [[TMP3:%.*]] = load i32*, i32** [[LOOPVAR_ADDR]], align 8
183 // CHECK-NEXT:    store i32 [[CALL]], i32* [[TMP3]], align 4
184 // CHECK-NEXT:    ret void
185 //
186