teams_distribute_parallel_for_simd_reduction_codegen.cpp 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348
  1. // RUN: %clang_cc1 -DCHECK -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %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 %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 %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 %s --check-prefix CHECK --check-prefix CHECK-32
  7. // 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 %s --check-prefix LAMBDA --check-prefix LAMBDA-64
  8. // 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
  9. // 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 %s --check-prefix LAMBDA --check-prefix LAMBDA-64
  10. // expected-no-diagnostics
  11. #ifndef HEADER
  12. #define HEADER
  13. template <typename T>
  14. T tmain() {
  15. T t_var = T();
  16. T vec[] = {1, 2};
  17. #pragma omp target
  18. #pragma omp teams distribute parallel for simd reduction(+: t_var)
  19. for (int i = 0; i < 2; ++i) {
  20. t_var += (T) i;
  21. }
  22. return T();
  23. }
  24. int main() {
  25. static int sivar;
  26. #ifdef LAMBDA
  27. // LAMBDA: [[RED_VAR:@.+]] = common global [8 x {{.+}}] zeroinitializer
  28. // LAMBDA-LABEL: @main
  29. // LAMBDA: call void [[OUTER_LAMBDA:@.+]](
  30. [&]() {
  31. // LAMBDA: define{{.*}} internal{{.*}} void [[OUTER_LAMBDA]](
  32. // LAMBDA: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 1, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0)
  33. // LAMBDA: call void @[[LOFFL1:.+]](
  34. // LAMBDA: ret
  35. #pragma omp target
  36. #pragma omp teams distribute parallel for simd reduction(+: sivar)
  37. for (int i = 0; i < 2; ++i) {
  38. // LAMBDA: define{{.*}} internal{{.*}} void @[[LOFFL1]](i{{64|32}} [[SIVAR_ARG:%.+]])
  39. // LAMBDA: [[SIVAR_ADDR:%.+]] = alloca i{{.+}},
  40. // LAMBDA: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
  41. // LAMBDA: [[SIVAR_CONV:%.+]] = bitcast{{.+}} [[SIVAR_ADDR]] to
  42. // LAMBDA: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[LOUTL1:.+]] to {{.+}}, {{.+}} [[SIVAR_CONV]])
  43. // LAMBDA: ret void
  44. // LAMBDA: define internal void @[[LOUTL1]]({{.+}}, {{.+}}, {{.+}} [[SIVAR_ARG:%.+]])
  45. // Skip global and bound tid vars
  46. // LAMBDA: {{.+}} = alloca i32*,
  47. // LAMBDA: {{.+}} = alloca i32*,
  48. // LAMBDA: [[SIVAR_ADDR:%.+]] = alloca i{{.+}}*,
  49. // LAMBDA: [[SIVAR_PRIV:%.+]] = alloca i{{.+}},
  50. // LAMBDA: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
  51. // LAMBDA: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
  52. // LAMBDA: [[SIVAR_REF:%.+]] = load{{.+}}, {{.+}} [[SIVAR_ADDR]]
  53. // LAMBDA: store{{.+}} 0, {{.+}} [[SIVAR_PRIV]],
  54. // LAMBDA: call void @__kmpc_for_static_init_4(
  55. // LAMBDA: call void {{.*}} @__kmpc_fork_call({{.+}}, {{.+}}, {{.+}} @[[LPAR_OUTL:.+]] to
  56. // LAMBDA: call void @__kmpc_for_static_fini(
  57. // LAMBDA: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
  58. // LAMBDA: [[SIVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[SIVAR_PRIV]] to
  59. // LAMBDA: store{{.+}} [[SIVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
  60. // LAMBDA: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
  61. // LAMBDA: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce_nowait({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
  62. // LAMBDA: switch{{.+}} [[K_RED_RET]], label{{.+}} [
  63. // LAMBDA: {{.+}}, label %[[CASE1:.+]]
  64. // LAMBDA: {{.+}}, label %[[CASE2:.+]]
  65. // LAMBDA: ]
  66. // LAMBDA: [[CASE1]]:
  67. // LAMBDA-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_REF]],
  68. // LAMBDA-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  69. // LAMBDA-DAG: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], [[SIVAR_PRIV_VAL]]
  70. // LAMBDA: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_REF]],
  71. // LAMBDA: call void @__kmpc_end_reduce_nowait({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
  72. // LAMBDA: br
  73. // LAMBDA: [[CASE2]]:
  74. // LAMBDA-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  75. // LAMBDA-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_REF]], {{.+}} [[SIVAR_PRIV_VAL]]
  76. // LAMBDA: br
  77. // LAMBDA: define internal void @[[LPAR_OUTL]]({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[SIVAR_ARG:%.+]])
  78. // Skip global and bound tid vars, and prev lb and ub vars
  79. // LAMBDA: {{.+}} = alloca i32*,
  80. // LAMBDA: {{.+}} = alloca i32*,
  81. // LAMBDA: alloca i{{[0-9]+}},
  82. // LAMBDA: alloca i{{[0-9]+}},
  83. // LAMBDA: [[SIVAR_ADDR:%.+]] = alloca i{{.+}}*,
  84. // skip loop vars
  85. // LAMBDA: alloca i32,
  86. // LAMBDA: alloca i32,
  87. // LAMBDA: alloca i32,
  88. // LAMBDA: alloca i32,
  89. // LAMBDA: alloca i32,
  90. // LAMBDA: [[SIVAR_PRIV:%.+]] = alloca i{{.+}},
  91. // LAMBDA: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
  92. // LAMBDA: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
  93. // LAMBDA: [[SIVAR_REF:%.+]] = load{{.+}}, {{.+}} [[SIVAR_ADDR]]
  94. // LAMBDA: store{{.+}} 0, {{.+}} [[SIVAR_PRIV]],
  95. // LAMBDA: call void @__kmpc_for_static_init_4(
  96. // LAMBDA: store{{.+}}, {{.+}} [[SIVAR_PRIV]],
  97. // LAMBDA: call void [[INNER_LAMBDA:@.+]](
  98. // LAMBDA: call void @__kmpc_for_static_fini(
  99. // LAMBDA: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
  100. // LAMBDA: [[SIVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[SIVAR_PRIV]] to
  101. // LAMBDA: store{{.+}} [[SIVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
  102. // LAMBDA: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
  103. // LAMBDA: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce_nowait({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
  104. // LAMBDA: switch{{.+}} [[K_RED_RET]], label{{.+}} [
  105. // LAMBDA: {{.+}}, label %[[CASE1:.+]]
  106. // LAMBDA: {{.+}}, label %[[CASE2:.+]]
  107. // LAMBDA: ]
  108. // LAMBDA: [[CASE1]]:
  109. // LAMBDA-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_REF]],
  110. // LAMBDA-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  111. // LAMBDA-DAG: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], [[SIVAR_PRIV_VAL]]
  112. // LAMBDA: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_REF]],
  113. // LAMBDA: call void @__kmpc_end_reduce_nowait({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
  114. // LAMBDA: br
  115. // LAMBDA: [[CASE2]]:
  116. // LAMBDA-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  117. // LAMBDA-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_REF]], {{.+}} [[SIVAR_PRIV_VAL]]
  118. // LAMBDA: br
  119. sivar += i;
  120. [&]() {
  121. // LAMBDA: define {{.+}} void [[INNER_LAMBDA]](%{{.+}}* [[ARG_PTR:%.+]])
  122. // LAMBDA: store %{{.+}}* [[ARG_PTR]], %{{.+}}** [[ARG_PTR_REF:%.+]],
  123. sivar += 4;
  124. // LAMBDA: [[ARG_PTR:%.+]] = load %{{.+}}*, %{{.+}}** [[ARG_PTR_REF]]
  125. // LAMBDA: [[SIVAR_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
  126. // LAMBDA: [[SIVAR_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[SIVAR_PTR_REF]]
  127. // LAMBDA: [[SIVAR_VAL:%.+]] = load i{{[0-9]+}}, i{{[0-9]+}}* [[SIVAR_REF]]
  128. // LAMBDA: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], 4
  129. // LAMBDA: store i{{[0-9]+}} [[SIVAR_INC]], i{{[0-9]+}}* [[SIVAR_REF]]
  130. }();
  131. }
  132. }();
  133. return 0;
  134. // LAMBDA: !{!"llvm.loop.vectorize.enable", i1 true}
  135. #else
  136. #pragma omp target
  137. #pragma omp teams distribute parallel for simd reduction(+: sivar)
  138. for (int i = 0; i < 2; ++i) {
  139. sivar += i;
  140. }
  141. return tmain<int>();
  142. #endif
  143. }
  144. // CHECK: [[RED_VAR:@.+]] = common global [8 x {{.+}}] zeroinitializer
  145. // CHECK: define {{.*}}i{{[0-9]+}} @main()
  146. // CHECK: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 1, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0)
  147. // CHECK: call void @[[OFFL1:.+]](i{{64|32}} %{{.+}})
  148. // CHECK: {{%.+}} = call{{.*}} i32 @[[TMAIN_INT:.+]]()
  149. // CHECK: ret
  150. // CHECK: define{{.*}} void @[[OFFL1]](i{{64|32}} [[SIVAR_ARG:%.+]])
  151. // CHECK: [[SIVAR_ADDR:%.+]] = alloca i{{.+}},
  152. // CHECK: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
  153. // CHECK-64: [[SIVAR_CONV:%.+]] = bitcast{{.+}} [[SIVAR_ADDR]] to
  154. // CHECK-64: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}} [[SIVAR_CONV]])
  155. // CHECK-32: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}} [[SIVAR_ADDR]])
  156. // CHECK: ret void
  157. // CHECK: define internal void @[[OUTL1]]({{.+}}, {{.+}}, {{.+}} [[SIVAR_ARG:%.+]])
  158. // Skip global and bound tid vars
  159. // CHECK: {{.+}} = alloca i32*,
  160. // CHECK: {{.+}} = alloca i32*,
  161. // CHECK: [[SIVAR_ADDR:%.+]] = alloca i{{.+}}*,
  162. // CHECK: [[SIVAR_PRIV:%.+]] = alloca i{{.+}},
  163. // CHECK: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
  164. // CHECK: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
  165. // CHECK: [[SIVAR_REF:%.+]] = load{{.+}}, {{.+}} [[SIVAR_ADDR]]
  166. // CHECK: store{{.+}} 0, {{.+}} [[SIVAR_PRIV]],
  167. // CHECK: call void @__kmpc_for_static_init_4(
  168. // CHECK: call void {{.*}} @__kmpc_fork_call({{.+}}, {{.+}}, {{.+}} @[[PAR_OUTL:.+]] to
  169. // CHECK: call void @__kmpc_for_static_fini(
  170. // CHECK: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
  171. // CHECK: [[SIVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[SIVAR_PRIV]] to
  172. // CHECK: store{{.+}} [[SIVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
  173. // CHECK: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
  174. // CHECK: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce_nowait({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
  175. // CHECK: switch{{.+}} [[K_RED_RET]], label{{.+}} [
  176. // CHECK: {{.+}}, label %[[CASE1:.+]]
  177. // CHECK: {{.+}}, label %[[CASE2:.+]]
  178. // CHECK: ]
  179. // CHECK: [[CASE1]]:
  180. // CHECK-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_REF]],
  181. // CHECK-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  182. // CHECK-DAG: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], [[SIVAR_PRIV_VAL]]
  183. // CHECK: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_REF]],
  184. // CHECK: call void @__kmpc_end_reduce_nowait({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
  185. // CHECK: br
  186. // CHECK: [[CASE2]]:
  187. // CHECK-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  188. // CHECK-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_REF]], {{.+}} [[SIVAR_PRIV_VAL]]
  189. // CHECK: br
  190. // CHECK: define internal void @[[PAR_OUTL]]({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[SIVAR_ARG:%.+]])
  191. // Skip global and bound tid vars, and prev lb and ub
  192. // CHECK: {{.+}} = alloca i32*,
  193. // CHECK: {{.+}} = alloca i32*,
  194. // CHECK: alloca i{{[0-9]+}},
  195. // CHECK: alloca i{{[0-9]+}},
  196. // CHECK: [[SIVAR_ADDR:%.+]] = alloca i{{.+}}*,
  197. // skip loop vars
  198. // CHECK: alloca i32,
  199. // CHECK: alloca i32,
  200. // CHECK: alloca i32,
  201. // CHECK: alloca i32,
  202. // CHECK: alloca i32,
  203. // CHECK: [[SIVAR_PRIV:%.+]] = alloca i{{.+}},
  204. // CHECK: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
  205. // CHECK: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
  206. // CHECK: [[SIVAR_REF:%.+]] = load{{.+}}, {{.+}} [[SIVAR_ADDR]]
  207. // CHECK: store{{.+}} 0, {{.+}} [[SIVAR_PRIV]],
  208. // CHECK: call void @__kmpc_for_static_init_4(
  209. // CHECK: store{{.+}}, {{.+}} [[SIVAR_PRIV]],
  210. // CHECK: call void @__kmpc_for_static_fini(
  211. // CHECK: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
  212. // CHECK: [[SIVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[SIVAR_PRIV]] to
  213. // CHECK: store{{.+}} [[SIVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
  214. // CHECK: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
  215. // CHECK: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce_nowait({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
  216. // CHECK: switch{{.+}} [[K_RED_RET]], label{{.+}} [
  217. // CHECK: {{.+}}, label %[[CASE1:.+]]
  218. // CHECK: {{.+}}, label %[[CASE2:.+]]
  219. // CHECK: ]
  220. // CHECK: [[CASE1]]:
  221. // CHECK-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_REF]],
  222. // CHECK-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  223. // CHECK-DAG: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], [[SIVAR_PRIV_VAL]]
  224. // CHECK: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_REF]],
  225. // CHECK: call void @__kmpc_end_reduce_nowait({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
  226. // CHECK: br
  227. // CHECK: [[CASE2]]:
  228. // CHECK-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
  229. // CHECK-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_REF]], {{.+}} [[SIVAR_PRIV_VAL]]
  230. // CHECK: br
  231. // CHECK: define{{.*}} i{{[0-9]+}} @[[TMAIN_INT]]()
  232. // CHECK: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 1,
  233. // CHECK: call void @[[TOFFL1:.+]]({{.+}})
  234. // CHECK: ret
  235. // CHECK: define{{.*}} void @[[TOFFL1]](i{{64|32}} [[TVAR_ARG:%.+]])
  236. // CHECK: [[TVAR_ADDR:%.+]] = alloca i{{.+}},
  237. // CHECK: store{{.+}} [[TVAR_ARG]], {{.+}} [[TVAR_ADDR]],
  238. // CHECK-64: [[TVAR_CONV:%.+]] = bitcast{{.+}} [[TVAR_ADDR]] to
  239. // CHECK-64: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[TOUTL1:.+]] to {{.+}}, {{.+}} [[TVAR_CONV]])
  240. // CHECK-32: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[TOUTL1:.+]] to {{.+}}, {{.+}} [[TVAR_ADDR]])
  241. // CHECK: ret void
  242. // CHECK: define internal void @[[TOUTL1]]({{.+}}, {{.+}}, {{.+}} [[TVAR_ARG:%.+]])
  243. // Skip global and bound tid vars
  244. // CHECK: {{.+}} = alloca i32*,
  245. // CHECK: {{.+}} = alloca i32*,
  246. // CHECK: [[TVAR_ADDR:%.+]] = alloca i{{.+}}*,
  247. // CHECK: [[TVAR_PRIV:%.+]] = alloca i{{.+}},
  248. // CHECK: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
  249. // CHECK: store{{.+}} [[TVAR_ARG]], {{.+}} [[TVAR_ADDR]],
  250. // CHECK: [[TVAR_REF:%.+]] = load{{.+}}, {{.+}} [[TVAR_ADDR]]
  251. // CHECK: store{{.+}} 0, {{.+}} [[TVAR_PRIV]],
  252. // CHECK: call void @__kmpc_for_static_init_4(
  253. // CHECK: call void {{.*}} @__kmpc_fork_call({{.+}}, {{.+}}, {{.+}} @[[TPAR_OUTL:.+]] to
  254. // CHECK: call void @__kmpc_for_static_fini(
  255. // CHECK: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
  256. // CHECK: [[TVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[TVAR_PRIV]] to
  257. // CHECK: store{{.+}} [[TVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
  258. // CHECK: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
  259. // CHECK: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce_nowait({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
  260. // CHECK: switch{{.+}} [[K_RED_RET]], label{{.+}} [
  261. // CHECK: {{.+}}, label %[[CASE1:.+]]
  262. // CHECK: {{.+}}, label %[[CASE2:.+]]
  263. // CHECK: ]
  264. // CHECK: [[CASE1]]:
  265. // CHECK-DAG: [[TVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_REF]],
  266. // CHECK-DAG: [[TVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_PRIV]],
  267. // CHECK-DAG: [[TVAR_INC:%.+]] = add{{.+}} [[TVAR_VAL]], [[TVAR_PRIV_VAL]]
  268. // CHECK: store{{.+}} [[TVAR_INC]], {{.+}} [[TVAR_REF]],
  269. // CHECK: call void @__kmpc_end_reduce_nowait({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
  270. // CHECK: br
  271. // CHECK: [[CASE2]]:
  272. // CHECK-DAG: [[TVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_PRIV]],
  273. // CHECK-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[TVAR_REF]], {{.+}} [[TVAR_PRIV_VAL]]
  274. // CHECK: br
  275. // CHECK: define internal void @[[TPAR_OUTL]]({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[TVAR_ARG:%.+]])
  276. // Skip global and bound tid vars, and prev lb and ub vars
  277. // CHECK: {{.+}} = alloca i32*,
  278. // CHECK: {{.+}} = alloca i32*,
  279. // CHECK: alloca i{{[0-9]+}},
  280. // CHECK: alloca i{{[0-9]+}},
  281. // CHECK: [[TVAR_ADDR:%.+]] = alloca i{{.+}}*,
  282. // skip loop vars
  283. // CHECK: alloca i32,
  284. // CHECK: alloca i32,
  285. // CHECK: alloca i32,
  286. // CHECK: alloca i32,
  287. // CHECK: alloca i32,
  288. // CHECK: [[TVAR_PRIV:%.+]] = alloca i{{.+}},
  289. // CHECK: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
  290. // CHECK: store{{.+}} [[TVAR_ARG]], {{.+}} [[TVAR_ADDR]],
  291. // CHECK: [[TVAR_REF:%.+]] = load{{.+}}, {{.+}} [[TVAR_ADDR]]
  292. // CHECK: store{{.+}} 0, {{.+}} [[TVAR_PRIV]],
  293. // CHECK: call void @__kmpc_for_static_init_4(
  294. // CHECK: store{{.+}}, {{.+}} [[TVAR_PRIV]],
  295. // CHECK: call void @__kmpc_for_static_fini(
  296. // CHECK: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
  297. // CHECK: [[TVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[TVAR_PRIV]] to
  298. // CHECK: store{{.+}} [[TVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
  299. // CHECK: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
  300. // CHECK: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce_nowait({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
  301. // CHECK: switch{{.+}} [[K_RED_RET]], label{{.+}} [
  302. // CHECK: {{.+}}, label %[[CASE1:.+]]
  303. // CHECK: {{.+}}, label %[[CASE2:.+]]
  304. // CHECK: ]
  305. // CHECK: [[CASE1]]:
  306. // CHECK-DAG: [[TVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_REF]],
  307. // CHECK-DAG: [[TVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_PRIV]],
  308. // CHECK-DAG: [[TVAR_INC:%.+]] = add{{.+}} [[TVAR_VAL]], [[TVAR_PRIV_VAL]]
  309. // CHECK: store{{.+}} [[TVAR_INC]], {{.+}} [[TVAR_REF]],
  310. // CHECK: call void @__kmpc_end_reduce_nowait({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
  311. // CHECK: br
  312. // CHECK: [[CASE2]]:
  313. // CHECK-DAG: [[TVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_PRIV]],
  314. // CHECK-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[TVAR_REF]], {{.+}} [[TVAR_PRIV_VAL]]
  315. // CHECK: br
  316. // CHECK: !{!"llvm.loop.vectorize.enable", i1 true}
  317. #endif