test/OpenMP/target_teams_distribute_simd_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
  29 struct 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
  36 volatile int g = 1212;
  37 volatile int &g1 = g;
  38
  39 template <class T>
  40 struct 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
  52 template <typename T>
  53 T 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 simd 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,
  68 S<float> test;
  69 // CHECK-DAG: [[T_VAR:@.+]] = global i{{[0-9]+}} 333,
  70 int t_var = 333;
  71 // CHECK-DAG: [[VEC:@.+]] = global [2 x i{{[0-9]+}}] [i{{[0-9]+}} 1, i{{[0-9]+}} 2],
  72 int vec[] = {1, 2};
  73 // CHECK-DAG: [[S_ARR:@.+]] = global [2 x [[S_FLOAT_TY]]] zeroinitializer,
  74 S<float> s_arr[] = {1, 2};
  75 // CHECK-DAG: [[VAR:@.+]] = global [[S_FLOAT_TY]] zeroinitializer,
  76 S<float> var(3);
  77 // CHECK-DAG: [[SIVAR:@.+]] = internal global i{{[0-9]+}} 0,
  78
  79 int 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 simd 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 simd 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