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