test/OpenMP/for_private_codegen.cpp

   1 // RUN: %clang_cc1 -verify -fopenmp -x c++ -triple x86_64-unknown-unknown -emit-llvm %s -o - | FileCheck %s
   2 // RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -triple x86_64-unknown-unknown -emit-pch -o %t %s
   3 // RUN: %clang_cc1 -fopenmp -x c++ -triple x86_64-unknown-unknown -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
   4 // RUN: %clang_cc1 -verify -fopenmp -x c++ -std=c++11 -DLAMBDA -triple %itanium_abi_triple -emit-llvm %s -o - | FileCheck -check-prefix=LAMBDA %s
   5 // RUN: %clang_cc1 -verify -fopenmp -x c++ -fblocks -DBLOCKS -triple %itanium_abi_triple -emit-llvm %s -o - | FileCheck -check-prefix=BLOCKS %s
   6 // expected-no-diagnostics
   7 #ifndef HEADER
   8 #define HEADER
   9
  10 template <class T>
  11 struct S {
  12   T f;
  13   S(T a) : f(a) {}
  14   S() : f() {}
  15   operator T() { return T(); }
  16   ~S() {}
  17 };
  18
  19 volatile double g;
  20
  21 // CHECK: [[S_FLOAT_TY:%.+]] = type { float }
  22 // CHECK: [[CAP_MAIN_TY:%.+]] = type { i{{[0-9]+}}*, [2 x i{{[0-9]+}}]*, [2 x [[S_FLOAT_TY]]]*, [[S_FLOAT_TY]]* }
  23 // CHECK: [[S_INT_TY:%.+]] = type { i{{[0-9]+}} }
  24 // CHECK: [[CAP_TMAIN_TY:%.+]] = type { i{{[0-9]+}}*, [2 x i{{[0-9]+}}]*, [2 x [[S_INT_TY]]]*, [[S_INT_TY]]* }
  25 template <typename T>
  26 T tmain() {
  27   S<T> test;
  28   T t_var = T();
  29   T vec[] = {1, 2};
  30   S<T> s_arr[] = {1, 2};
  31   S<T> var(3);
  32 #pragma omp parallel
  33 #pragma omp for private(t_var, vec, s_arr, s_arr, var, var)
  34   for (int i = 0; i < 2; ++i) {
  35     vec[i] = t_var;
  36     s_arr[i] = var;
  37   }
  38   return T();
  39 }
  40
  41 int main() {
  42 #ifdef LAMBDA
  43   // LAMBDA: [[G:@.+]] = global double
  44   // LAMBDA-LABEL: @main
  45   // LAMBDA: call{{( x86_thiscallcc)?}} void [[OUTER_LAMBDA:@.+]](
  46   [&]() {
  47   // LAMBDA: define{{.*}} internal{{.*}} void [[OUTER_LAMBDA]](
  48   // LAMBDA: call void {{.+}} @__kmpc_fork_call({{.+}}, i32 1, {{.+}}* [[OMP_REGION:@.+]] to {{.+}}, i8* %{{.+}})
  49 #pragma omp parallel
  50 #pragma omp for private(g)
  51   for (int i = 0; i < 2; ++i) {
  52     // LAMBDA: define{{.*}} internal{{.*}} void [[OMP_REGION]](i32* %{{.+}}, i32* %{{.+}}, %{{.+}}* [[ARG:%.+]])
  53     // LAMBDA: [[G_PRIVATE_ADDR:%.+]] = alloca double,
  54     // LAMBDA: store %{{.+}}* [[ARG]], %{{.+}}** [[ARG_REF:%.+]],
  55     g = 1;
  56     // LAMBDA: call void @__kmpc_for_static_init_4(
  57     // LAMBDA: store volatile double 1.0{{.+}}, double* [[G_PRIVATE_ADDR]],
  58     // LAMBDA: [[G_PRIVATE_ADDR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG:%.+]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
  59     // LAMBDA: store double* [[G_PRIVATE_ADDR]], double** [[G_PRIVATE_ADDR_REF]]
  60     // LAMBDA: call{{( x86_thiscallcc)?}} void [[INNER_LAMBDA:@.+]](%{{.+}}* [[ARG]])
  61     // LAMBDA: call void @__kmpc_for_static_fini(
  62     [&]() {
  63       // LAMBDA: define {{.+}} void [[INNER_LAMBDA]](%{{.+}}* [[ARG_PTR:%.+]])
  64       // LAMBDA: store %{{.+}}* [[ARG_PTR]], %{{.+}}** [[ARG_PTR_REF:%.+]],
  65       g = 2;
  66       // LAMBDA: [[ARG_PTR:%.+]] = load %{{.+}}*, %{{.+}}** [[ARG_PTR_REF]]
  67       // LAMBDA: [[G_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
  68       // LAMBDA: [[G_REF:%.+]] = load double*, double** [[G_PTR_REF]]
  69       // LAMBDA: store volatile double 2.0{{.+}}, double* [[G_REF]]
  70     }();
  71   }
  72   }();
  73   return 0;
  74 #elif defined(BLOCKS)
  75   // BLOCKS: [[G:@.+]] = global double
  76   // BLOCKS-LABEL: @main
  77   // BLOCKS: call void {{%.+}}(i8
  78   ^{
  79   // BLOCKS: define{{.*}} internal{{.*}} void {{.+}}(i8*
  80   // BLOCKS: call void {{.+}} @__kmpc_fork_call({{.+}}, i32 1, {{.+}}* [[OMP_REGION:@.+]] to {{.+}}, i8* {{.+}})
  81 #pragma omp parallel
  82 #pragma omp for private(g)
  83   for (int i = 0; i < 2; ++i) {
  84     // BLOCKS: define{{.*}} internal{{.*}} void [[OMP_REGION]](i32* %{{.+}}, i32* %{{.+}}, %{{.+}}* [[ARG:%.+]])
  85     // BLOCKS: [[G_PRIVATE_ADDR:%.+]] = alloca double,
  86     // BLOCKS: store %{{.+}}* [[ARG]], %{{.+}}** [[ARG_REF:%.+]],
  87     g = 1;
  88     // BLOCKS: call void @__kmpc_for_static_init_4(
  89     // BLOCKS: store volatile double 1.0{{.+}}, double* [[G_PRIVATE_ADDR]],
  90     // BLOCKS-NOT: [[G]]{{[[^:word:]]}}
  91     // BLOCKS: double* [[G_PRIVATE_ADDR]]
  92     // BLOCKS-NOT: [[G]]{{[[^:word:]]}}
  93     // BLOCKS: call void {{%.+}}(i8
  94     // BLOCKS: call void @__kmpc_for_static_fini(
  95     ^{
  96       // BLOCKS: define {{.+}} void {{@.+}}(i8*
  97       g = 2;
  98       // BLOCKS-NOT: [[G]]{{[[^:word:]]}}
  99       // BLOCKS: store volatile double 2.0{{.+}}, double*
 100       // BLOCKS-NOT: [[G]]{{[[^:word:]]}}
 101       // BLOCKS: ret
 102     }();
 103   }
 104   }();
 105   return 0;
 106 #else
 107   S<float> test;
 108   int t_var = 0;
 109   int vec[] = {1, 2};
 110   S<float> s_arr[] = {1, 2};
 111   S<float> var(3);
 112 #pragma omp parallel
 113 #pragma omp for private(t_var, vec, s_arr, s_arr, var, var)
 114   for (int i = 0; i < 2; ++i) {
 115     vec[i] = t_var;
 116     s_arr[i] = var;
 117   }
 118   int i;
 119 #pragma omp parallel
 120 #pragma omp for private(i)
 121   for (i = 0; i < 2; ++i) {
 122     ;
 123   }
 124   return tmain<int>();
 125 #endif
 126 }
 127
 128 // CHECK: define i{{[0-9]+}} @main()
 129 // CHECK: [[TEST:%.+]] = alloca [[S_FLOAT_TY]],
 130 // CHECK: call {{.*}} [[S_FLOAT_TY_DEF_CONSTR:@.+]]([[S_FLOAT_TY]]* [[TEST]])
 131 // CHECK: %{{.+}} = bitcast [[CAP_MAIN_TY]]*
 132 // CHECK: call void (%{{.+}}*, i{{[0-9]+}}, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)*, ...) @__kmpc_fork_call(%{{.+}}* @{{.+}}, i{{[0-9]+}} 1, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)* bitcast (void (i{{[0-9]+}}*, i{{[0-9]+}}*, [[CAP_MAIN_TY]]*)* [[MAIN_MICROTASK:@.+]] to void
 133 // CHECK: = call i{{.+}} [[TMAIN_INT:@.+]]()
 134 // CHECK: call void [[S_FLOAT_TY_DESTR:@.+]]([[S_FLOAT_TY]]*
 135 // CHECK: ret
 136 //
 137 // CHECK: define internal void [[MAIN_MICROTASK]](i{{[0-9]+}}* [[GTID_ADDR:%.+]], i{{[0-9]+}}* %{{.+}}, [[CAP_MAIN_TY]]* %{{.+}})
 138 // CHECK: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
 139 // CHECK: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
 140 // CHECK: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_FLOAT_TY]]],
 141 // CHECK-NOT: alloca [2 x [[S_FLOAT_TY]]],
 142 // CHECK: [[VAR_PRIV:%.+]] = alloca [[S_FLOAT_TY]],
 143 // CHECK-NOT: alloca [[S_FLOAT_TY]],
 144 // CHECK: store i{{[0-9]+}}* [[GTID_ADDR]], i{{[0-9]+}}** [[GTID_ADDR_REF:%.+]]
 145 // CHECK-NOT: [[T_VAR_PRIV]]
 146 // CHECK-NOT: [[VEC_PRIV]]
 147 // CHECK: {{.+}}:
 148 // CHECK: [[S_ARR_PRIV_ITEM:%.+]] = phi [[S_FLOAT_TY]]*
 149 // CHECK: call {{.*}} [[S_FLOAT_TY_DEF_CONSTR]]([[S_FLOAT_TY]]* [[S_ARR_PRIV_ITEM]])
 150 // CHECK-NOT: [[T_VAR_PRIV]]
 151 // CHECK-NOT: [[VEC_PRIV]]
 152 // CHECK: call {{.*}} [[S_FLOAT_TY_DEF_CONSTR]]([[S_FLOAT_TY]]* [[VAR_PRIV]])
 153 // CHECK: call void @__kmpc_for_static_init_4(
 154 // CHECK: call void @__kmpc_for_static_fini(
 155 // CHECK-DAG: call void [[S_FLOAT_TY_DESTR]]([[S_FLOAT_TY]]* [[VAR_PRIV]])
 156 // CHECK-DAG: call void [[S_FLOAT_TY_DESTR]]([[S_FLOAT_TY]]*
 157 // CHECK: ret void
 158
 159 // CHECK: define {{.*}} i{{[0-9]+}} [[TMAIN_INT]]()
 160 // CHECK: [[TEST:%.+]] = alloca [[S_INT_TY]],
 161 // CHECK: call {{.*}} [[S_INT_TY_DEF_CONSTR:@.+]]([[S_INT_TY]]* [[TEST]])
 162 // CHECK: call void (%{{.+}}*, i{{[0-9]+}}, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)*, ...) @__kmpc_fork_call(%{{.+}}* @{{.+}}, i{{[0-9]+}} 1, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)* bitcast (void (i{{[0-9]+}}*, i{{[0-9]+}}*, [[CAP_TMAIN_TY]]*)* [[TMAIN_MICROTASK:@.+]] to void
 163 // CHECK: call void [[S_INT_TY_DESTR:@.+]]([[S_INT_TY]]*
 164 // CHECK: ret
 165 //
 166 // CHECK: define internal void [[TMAIN_MICROTASK]](i{{[0-9]+}}* [[GTID_ADDR:%.+]], i{{[0-9]+}}* %{{.+}}, [[CAP_TMAIN_TY]]* %{{.+}})
 167 // CHECK: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
 168 // CHECK: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
 169 // CHECK: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_INT_TY]]],
 170 // CHECK-NOT: alloca [2 x [[S_INT_TY]]],
 171 // CHECK: [[VAR_PRIV:%.+]] = alloca [[S_INT_TY]],
 172 // CHECK-NOT: alloca [[S_INT_TY]],
 173 // CHECK: store i{{[0-9]+}}* [[GTID_ADDR]], i{{[0-9]+}}** [[GTID_ADDR_REF:%.+]]
 174 // CHECK-NOT: [[T_VAR_PRIV]]
 175 // CHECK-NOT: [[VEC_PRIV]]
 176 // CHECK: {{.+}}:
 177 // CHECK: [[S_ARR_PRIV_ITEM:%.+]] = phi [[S_INT_TY]]*
 178 // CHECK: call {{.*}} [[S_INT_TY_DEF_CONSTR]]([[S_INT_TY]]* [[S_ARR_PRIV_ITEM]])
 179 // CHECK-NOT: [[T_VAR_PRIV]]
 180 // CHECK-NOT: [[VEC_PRIV]]
 181 // CHECK: call {{.*}} [[S_INT_TY_DEF_CONSTR]]([[S_INT_TY]]* [[VAR_PRIV]])
 182 // CHECK: call void @__kmpc_for_static_init_4(
 183 // CHECK: call void @__kmpc_for_static_fini(
 184 // CHECK-DAG: call void [[S_INT_TY_DESTR]]([[S_INT_TY]]* [[VAR_PRIV]])
 185 // CHECK-DAG: call void [[S_INT_TY_DESTR]]([[S_INT_TY]]*
 186 // CHECK: ret void
 187 #endif
 188