Clang Project

clang_source_code/test/OpenMP/target_teams_distribute_private_codegen.cpp
1// RUN: %clang_cc1 -DCHECK -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck -allow-deprecated-dag-overlap  %s --check-prefix CHECK --check-prefix CHECK-64
2// RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
3// RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck -allow-deprecated-dag-overlap  %s --check-prefix CHECK --check-prefix CHECK-64
4// RUN: %clang_cc1 -DCHECK -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck -allow-deprecated-dag-overlap  %s --check-prefix CHECK --check-prefix CHECK-32
5// RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
6// RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck -allow-deprecated-dag-overlap  %s --check-prefix CHECK --check-prefix CHECK-32
7
8// RUN: %clang_cc1 -DCHECK -verify -fopenmp-simd -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck -allow-deprecated-dag-overlap  --check-prefix SIMD-ONLY0 %s
9// RUN: %clang_cc1 -DCHECK -fopenmp-simd -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
10// RUN: %clang_cc1 -DCHECK -fopenmp-simd -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck -allow-deprecated-dag-overlap  --check-prefix SIMD-ONLY0 %s
11// RUN: %clang_cc1 -DCHECK -verify -fopenmp-simd -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck -allow-deprecated-dag-overlap  --check-prefix SIMD-ONLY0 %s
12// RUN: %clang_cc1 -DCHECK -fopenmp-simd -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
13// RUN: %clang_cc1 -DCHECK -fopenmp-simd -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck -allow-deprecated-dag-overlap  --check-prefix SIMD-ONLY0 %s
14// SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
15
16// RUN: %clang_cc1 -DLAMBDA -verify -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck -allow-deprecated-dag-overlap  %s --check-prefix LAMBDA --check-prefix LAMBDA-64
17// RUN: %clang_cc1 -DLAMBDA -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
18// RUN: %clang_cc1 -DLAMBDA -fopenmp -x c++  -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck -allow-deprecated-dag-overlap  %s --check-prefix LAMBDA --check-prefix LAMBDA-64
19
20// RUN: %clang_cc1 -DLAMBDA -verify -fopenmp-simd -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck -allow-deprecated-dag-overlap  --check-prefix SIMD-ONLY1 %s
21// RUN: %clang_cc1 -DLAMBDA -fopenmp-simd -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
22// RUN: %clang_cc1 -DLAMBDA -fopenmp-simd -x c++  -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck -allow-deprecated-dag-overlap  --check-prefix SIMD-ONLY1 %s
23// SIMD-ONLY1-NOT: {{__kmpc|__tgt}}
24
25// expected-no-diagnostics
26#ifndef HEADER
27#define HEADER
28
29struct St {
30  int a, b;
31  St() : a(0), b(0) {}
32  St(const St &st) : a(st.a + st.b), b(0) {}
33  ~St() {}
34};
35
36volatile int g = 1212;
37volatile int &g1 = g;
38
39template <class T>
40struct S {
41  T f;
42  S(T a) : f(a + g) {}
43  S() : f(g) {}
44  S(const S &s, St t = St()) : f(s.f + t.a) {}
45  operator T() { return T(); }
46  ~S() {}
47};
48
49// CHECK-DAG: [[S_FLOAT_TY:%.+]] = type { float }
50// CHECK-DAG: [[S_INT_TY:%.+]] = type { i{{[0-9]+}} }
51
52template <typename T>
53T tmain() {
54  S<T> test;
55  T t_var = T();
56  T vec[] = {1, 2};
57  S<T> s_arr[] = {1, 2};
58  S<T> &var = test;
59#pragma omp target teams distribute private(t_var, vec, s_arr, var)
60  for (int i = 0; i < 2; ++i) {
61    vec[i] = t_var;
62    s_arr[i] = var;
63  }
64  return T();
65}
66
67// CHECK-DAG: [[TEST:@.+]] = global [[S_FLOAT_TY]] zeroinitializer,
68S<float> test;
69// CHECK-DAG: [[T_VAR:@.+]] = global i{{[0-9]+}} 333,
70int t_var = 333;
71// CHECK-DAG: [[VEC:@.+]] = global [2 x i{{[0-9]+}}] [i{{[0-9]+}} 1, i{{[0-9]+}} 2],
72int vec[] = {1, 2};
73// CHECK-DAG: [[S_ARR:@.+]] = global [2 x [[S_FLOAT_TY]]] zeroinitializer,
74S<float> s_arr[] = {1, 2};
75// CHECK-DAG: [[VAR:@.+]] = global [[S_FLOAT_TY]] zeroinitializer,
76S<float> var(3);
77// CHECK-DAG: [[SIVAR:@.+]] = internal global i{{[0-9]+}} 0,
78
79int main() {
80  static int sivar;
81#ifdef LAMBDA
82  // LAMBDA: [[G:@.+]] = global i{{[0-9]+}} 1212,
83  // LAMBDA-LABEL: @main
84  // LAMBDA: call void [[OUTER_LAMBDA:@.+]](
85  [&]() {
86    // LAMBDA: define{{.*}} internal{{.*}} void [[OUTER_LAMBDA]](
87    // LAMBDA: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 0, i8** null, i8** null, i{{64|32}}* null, i64* null, i32 0, i32 0)
88    // LAMBDA: call void @[[LOFFL1:.+]]()
89    // LAMBDA:  ret
90#pragma omp target teams distribute private(g, g1, sivar)
91  for (int i = 0; i < 2; ++i) {
92    // LAMBDA: define{{.*}} internal{{.*}} void @[[LOFFL1]]()
93    // LAMBDA: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}} @[[LOUTL1:.+]] to {{.+}})
94    // LAMBDA: ret void
95
96    // LAMBDA: define internal void @[[LOUTL1]]({{.+}})
97    // Skip global, bound tid and loop vars
98    // LAMBDA: {{.+}} = alloca i32*,
99    // LAMBDA: {{.+}} = alloca i32*,
100    // LAMBDA: alloca i32,
101    // LAMBDA: alloca i32,
102    // LAMBDA: alloca i32,
103    // LAMBDA: alloca i32,
104    // LAMBDA: alloca i32,
105    // LAMBDA: alloca i32,
106    // LAMBDA: [[G_PRIV:%.+]] = alloca i{{[0-9]+}},
107    // LAMBDA: [[G1_PRIV:%.+]] = alloca i{{[0-9]+}}
108    // LAMBDA: [[TMP:%.+]] = alloca i{{[0-9]+}}*,
109    // LAMBDA: [[SIVAR_PRIV:%.+]] = alloca i{{[0-9]+}},
110    // LAMBDA: store{{.+}} [[G1_PRIV]], {{.+}} [[TMP]],
111    g = 1;
112    g1 = 1;
113    sivar = 2;
114    // LAMBDA: call void @__kmpc_for_static_init_4(
115    // LAMBDA-DAG: store{{.+}} 1, {{.+}} [[G_PRIV]],
116    // LAMBDA-DAG: store{{.+}} 2, {{.+}} [[SIVAR_PRIV]],
117    // LAMBDA-DAG: [[G1_REF:%.+]] = load{{.+}}, {{.+}} [[TMP]],
118    // LAMBDA-DAG: store{{.+}} 1, {{.+}} [[G1_REF]],
119    // LAMBDA: call void [[INNER_LAMBDA:@.+]](
120    // LAMBDA: call void @__kmpc_for_static_fini(
121    [&]() {
122      // LAMBDA: define {{.+}} void [[INNER_LAMBDA]](%{{.+}}* [[ARG_PTR:%.+]])
123      // LAMBDA: store %{{.+}}* [[ARG_PTR]], %{{.+}}** [[ARG_PTR_REF:%.+]],
124      g = 2;
125      g1 = 2;
126      sivar = 4;
127      // LAMBDA: [[ARG_PTR:%.+]] = load %{{.+}}*, %{{.+}}** [[ARG_PTR_REF]]
128
129      // LAMBDA: [[G_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
130      // LAMBDA: [[G_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[G_PTR_REF]]
131      // LAMBDA: store i{{[0-9]+}} 2, i{{[0-9]+}}* [[G_REF]]
132      // LAMBDA: [[G1_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 1
133      // LAMBDA: [[G1_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[G1_PTR_REF]]
134      // LAMBDA: store i{{[0-9]+}} 2, i{{[0-9]+}}* [[G1_REF]]
135      // LAMBDA: [[SIVAR_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 2
136      // LAMBDA: [[SIVAR_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[SIVAR_PTR_REF]]
137      // LAMBDA: store i{{[0-9]+}} 4, i{{[0-9]+}}* [[SIVAR_REF]]
138    }();
139  }
140  }();
141  return 0;
142#else
143#pragma omp target teams distribute private(t_var, vec, s_arr, var, sivar)
144  for (int i = 0; i < 2; ++i) {
145    vec[i] = t_var;
146    s_arr[i] = var;
147    sivar += i;
148  }
149  return tmain<int>();
150#endif
151}
152
153// CHECK: define {{.*}}i{{[0-9]+}} @main()
154// CHECK: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 0, i8** null, i8** null, i{{64|32}}* null, i64* null, i32 0, i32 0)
155// CHECK: call void @[[OFFL1:.+]]()
156// CHECK: {{%.+}} = call{{.*}} i32 @[[TMAIN_INT:.+]]()
157// CHECK:  ret
158
159// CHECK: define{{.*}} void @[[OFFL1]]()
160// CHECK: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}} @[[OUTL1:.+]] to {{.+}})
161// CHECK: ret void
162
163// CHECK: define internal void @[[OUTL1]]({{.+}})
164// Skip global, bound tid and loop vars
165// CHECK: {{.+}} = alloca i32*,
166// CHECK: {{.+}} = alloca i32*,
167// CHECK: {{.+}} = alloca i32,
168// CHECK: {{.+}} = alloca i32,
169// CHECK: {{.+}} = alloca i32,
170// CHECK: {{.+}} = alloca i32,
171// CHECK: {{.+}} = alloca i32,
172// CHECK: {{.+}} = alloca i32,
173// CHECK-DAG: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
174// CHECK-DAG: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
175// CHECK-DAG: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_FLOAT_TY]]],
176// CHECK-DAG: [[VAR_PRIV:%.+]] = alloca [[S_FLOAT_TY]],
177// CHECK-DAG: [[SIVAR_PRIV:%.+]] = alloca i{{[0-9]+}},
178
179// private(s_arr)
180// CHECK-DAG: [[S_ARR_PRIV_BGN:%.+]] = getelementptr{{.*}} [2 x [[S_FLOAT_TY]]], [2 x [[S_FLOAT_TY]]]* [[S_ARR_PRIV]],
181// CHECK-DAG: [[S_ARR_PTR_ALLOC:%.+]] = phi{{.+}} [ [[S_ARR_PRIV_BGN]], {{.+}} ], [ [[S_ARR_NEXT:%.+]], {{.+}} ]
182// CHECK-DAG: call void @{{.+}}({{.+}} [[S_ARR_PTR_ALLOC]])
183// CHECK-DAG: [[S_ARR_NEXT]] = getelementptr {{.+}} [[S_ARR_PTR_ALLOC]],
184
185// private(var)
186// CHECK-DAG: call void @{{.+}}({{.+}} [[VAR_PRIV]])
187
188// CHECK: call void @__kmpc_for_static_init_4(
189// CHECK-DAG: {{.+}} = {{.+}} [[T_VAR_PRIV]]
190// CHECK-DAG: {{.+}} = {{.+}} [[VEC_PRIV]]
191// CHECK-DAG: {{.+}} = {{.+}} [[S_ARR_PRIV]]
192// CHECK-DAG: {{.+}} = {{.+}} [[VAR_PRIV]]
193// CHECK-DAG: {{.+}} = {{.+}} [[SIVAR_PRIV]]
194// CHECK: call void @__kmpc_for_static_fini(
195// CHECK: ret void
196
197
198// CHECK: define{{.*}} i{{[0-9]+}} @[[TMAIN_INT]]()
199// CHECK: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 0,
200// CHECK: call void @[[TOFFL1:.+]]()
201// CHECK:  ret
202
203// CHECK: define {{.*}}void @[[TOFFL1]]()
204// CHECK: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}} @[[TOUTL1:.+]] to {{.+}})
205// CHECK: ret void
206
207// CHECK: define internal void @[[TOUTL1]]({{.+}})
208// Skip global, bound tid and loop vars
209// CHECK: {{.+}} = alloca i32*,
210// CHECK: {{.+}} = alloca i32*,
211// CHECK: alloca i{{[0-9]+}},
212// CHECK: alloca i{{[0-9]+}},
213// CHECK: alloca i{{[0-9]+}},
214// CHECK: alloca i{{[0-9]+}},
215// CHECK: alloca i{{[0-9]+}},
216// CHECK: alloca i{{[0-9]+}},
217// CHECK: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
218// CHECK: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
219// CHECK: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_INT_TY]]],
220// CHECK: [[VAR_PRIV:%.+]] = alloca [[S_INT_TY]],
221// CHECK: [[TMP:%.+]] = alloca [[S_INT_TY]]*,
222
223// private(s_arr)
224// CHECK-DAG: [[S_ARR_PRIV_BGN:%.+]] = getelementptr{{.*}} [2 x [[S_INT_TY]]], [2 x [[S_INT_TY]]]* [[S_ARR_PRIV]],
225// CHECK-DAG: [[S_ARR_PTR_ALLOC:%.+]] = phi{{.+}} [ [[S_ARR_PRIV_BGN]], {{.+}} ], [ [[S_ARR_NEXT:%.+]], {{.+}} ]
226// CHECK-DAG: call void @{{.+}}({{.+}} [[S_ARR_PTR_ALLOC]])
227// CHECK-DAG: [[S_ARR_NEXT]] = getelementptr {{.+}} [[S_ARR_PTR_ALLOC]],
228
229// CHECK-DAG: [[S_ARR_PRIV_BGN:%.+]] = getelementptr{{.*}} [2 x [[S_INT_TY]]], [2 x [[S_INT_TY]]]* [[S_ARR_PRIV]],
230// CHECK-DAG: [[S_ARR_PTR_ALLOC:%.+]] = phi{{.+}} [ [[S_ARR_PRIV_BGN]], {{.+}} ], [ [[S_ARR_NEXT:%.+]], {{.+}} ]
231// CHECK-DAG: call void @{{.+}}({{.+}} [[S_ARR_PTR_ALLOC]])
232// CHECK-DAG: [[S_ARR_NEXT]] = getelementptr {{.+}} [[S_ARR_PTR_ALLOC]],
233
234// private(var)
235// CHECK-DAG: call void @{{.+}}({{.+}} [[VAR_PRIV]])
236// CHECK-DAG: store{{.+}} [[VAR_PRIV]], {{.+}} [[TMP]]
237
238// CHECK: call void @__kmpc_for_static_init_4(
239// CHECK-DAG: {{.+}} = {{.+}} [[T_VAR_PRIV]]
240// CHECK-DAG: {{.+}} = {{.+}} [[VEC_PRIV]]
241// CHECK-DAG: {{.+}} = {{.+}} [[S_ARR_PRIV]]
242// CHECK-DAG: {{.+}} = {{.+}} [[TMP]]
243// CHECK: call void @__kmpc_for_static_fini(
244// CHECK: ret void
245
246#endif
247