1 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s
2 // RUN: %clang_cc1 -fopenmp -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -std=c++11 -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -emit-pch -o %t %s
3 // RUN: %clang_cc1 -fopenmp -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
4 
5 // RUN: %clang_cc1 -verify -fopenmp-simd -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck --check-prefix SIMD-ONLY0 %s
6 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -std=c++11 -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -emit-pch -o %t %s
7 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
8 // SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
9 
10 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s
11 // RUN: %clang_cc1 -fopenmp -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -std=c++11 -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -emit-pch -o %t %s
12 // RUN: %clang_cc1 -fopenmp -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
13 
14 // RUN: %clang_cc1 -verify -fopenmp-simd -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck --check-prefix SIMD-ONLY0 %s
15 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -std=c++11 -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -emit-pch -o %t %s
16 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple %itanium_abi_triple -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
17 // SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
18 
19 // expected-no-diagnostics
20 #ifndef HEADER
21 #define HEADER
22 
23 typedef __INTPTR_TYPE__ intptr_t;
24 
25 // CHECK-DAG: [[IDENT_T_TY:%.+]] = type { i32, i32, i32, i32, i8* }
26 // CHECK-DAG: [[S_TY:%.+]] = type { [[INTPTR_T_TY:i[0-9]+]], [[INTPTR_T_TY]], [[INTPTR_T_TY]] }
27 // CHECK-DAG: [[STR:@.+]] = private unnamed_addr constant [23 x i8] c";unknown;unknown;0;0;;\00"
28 // CHECK-DAG: [[DEF_LOC_2:@.+]] = private unnamed_addr constant [[IDENT_T_TY]] { i32 0, i32 2, i32 0, i32 0, i8* getelementptr inbounds ([23 x i8], [23 x i8]* [[STR]], i32 0, i32 0) }
29 
30 void foo();
31 
32 struct S {
33   intptr_t a, b, c;
34   S(intptr_t a) : a(a) {}
35   operator char() { extern void mayThrow(); mayThrow(); return a; }
36   ~S() {}
37 };
38 
39 template <typename T, int C>
40 int tmain() {
41 #pragma omp target
42 #pragma omp teams
43 #pragma omp distribute parallel for num_threads(C)
44   for (int i = 0; i < 100; i++)
45     foo();
46 #pragma omp target
47 #pragma omp teams
48 #pragma omp distribute parallel for num_threads(T(23))
49   for (int i = 0; i < 100; i++)
50     foo();
51   return 0;
52 }
53 
54 int main() {
55   S s(0);
56   char a = s;
57 // CHECK: call i{{[0-9]+}} @__tgt_target_teams_mapper(
58 // CHECK: call void [[OFFLOADING_FUN_0:@.+]](
59 // CHECK: call i{{[0-9]+}} @__tgt_target_teams_mapper(
60 // CHECK: call void [[OFFLOADING_FUN_1:@.+]](
61 // CHECK: invoke{{.+}} [[TMAIN_5:@.+]]()
62 // CHECK: invoke{{.+}} [[TMAIN_1:@.+]]()
63 #pragma omp target
64 #pragma omp teams
65   // CHECK: define internal void [[OFFLOADING_FUN_0]](
66   // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[OMP_TEAMS_OUTLINED_0:@.+]] to {{.+}})
67 #pragma omp distribute parallel for num_threads(2)
68   for (int i = 0; i < 100; i++) {
69     // CHECK: define{{.+}} void [[OMP_TEAMS_OUTLINED_0]](
70     // CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 2)
71     // CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
72     foo();
73   }
74 #pragma omp target
75 #pragma omp teams
76   // CHECK: define internal void [[OFFLOADING_FUN_1]](
77 
78   // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}}* [[OMP_TEAMS_OUTLINED_1:@.+]] to {{.+}})
79 #pragma omp distribute parallel for num_threads(a)
80   for (int i = 0; i < 100; i++) {
81     // CHECK: define{{.+}} void [[OMP_TEAMS_OUTLINED_1]](
82     // CHECK-DAG: [[A_ADDR:%.+]] = alloca i8*,
83     // CHECK-DAG: [[A_REF:%.+]] = load i8*, i8** [[A_ADDR]],
84     // CHECK-DAG: [[A_VAL:%.+]] = load i8, i8* [[A_REF]],
85     // CHECK-DAG: [[A_EXT:%.+]] = sext i8 [[A_VAL]] to {{.+}}
86     // CHECK: call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 [[A_EXT]])
87     // CHECK: call {{.*}}void {{.*}} @__kmpc_fork_call(
88     foo();
89   }
90   return a + tmain<char, 5>() + tmain<S, 1>();
91 }
92 
93 // tmain 5
94 // CHECK-DAG: define {{.*}}i{{[0-9]+}} [[TMAIN_5]]()
95 // CHECK: call i{{[0-9]+}} @__tgt_target_teams_mapper(
96 // CHECK: call void [[T_OFFLOADING_FUN_0:@.+]](
97 // CHECK: call i{{[0-9]+}} @__tgt_target_teams_mapper(
98 // CHECK: call void [[T_OFFLOADING_FUN_1:@.+]](
99 
100 // tmain 1
101 // CHECK-DAG: define {{.*}}i{{[0-9]+}} [[TMAIN_1]]()
102 // CHECK: call i{{[0-9]+}} @__tgt_target_teams_mapper(
103 // CHECK: call void [[T_OFFLOADING_FUN_2:@.+]](
104 // CHECK: call i{{[0-9]+}} @__tgt_target_teams_mapper(
105 // CHECK: call void [[T_OFFLOADING_FUN_3:@.+]](
106 
107 // CHECK: define internal void [[T_OFFLOADING_FUN_0]](
108 // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_0:@.+]] to {{.+}})
109 
110 // CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_0]](
111 // CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 5)
112 // CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
113 
114 // CHECK: define internal void [[T_OFFLOADING_FUN_1]](
115 // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_1:@.+]] to {{.+}})
116 
117 // CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_1]](
118 // CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 23)
119 // CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
120 
121 // CHECK: define internal void [[T_OFFLOADING_FUN_2]](
122 // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_2:@.+]] to {{.+}})
123 
124 // CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_2]](
125 // CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 1)
126 // CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
127 
128 // CHECK: define internal void [[T_OFFLOADING_FUN_3]](
129 // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_3:@.+]] to {{.+}})
130 
131 // CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_3]](
132 // CHECK-DAG:   [[CALL_RES:%.+]] = invoke{{.*}} i8 [[S_TY_CHAR_OP:@.+]]([[S_TY]]* {{.+}})
133 // CHECK-DAG:   [[CALL_RES_SEXT:%.+]] = sext i8 [[CALL_RES]] to {{.+}}
134 // CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 [[CALL_RES_SEXT]])
135 // CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
136 #endif
137