xref: /llvm-project/clang/test/OpenMP/cancel_codegen.cpp (revision 2a6de8c32117e7d8f16798f62d0339e08d9f7cdb)
1 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -triple x86_64-apple-darwin13.4.0 -emit-llvm -o - %s | FileCheck %s
2 // RUN: %clang_cc1 -fopenmp -fopenmp-version=45 -x c++ -std=c++11 -triple x86_64-apple-darwin13.4.0 -emit-pch -o %t %s
3 // RUN: %clang_cc1 -fopenmp -fopenmp-version=45 -std=c++11 -include-pch %t -fsyntax-only -verify %s -triple x86_64-apple-darwin13.4.0 -emit-llvm -o - | FileCheck %s
4 // expected-no-diagnostics
5 #ifndef HEADER
6 #define HEADER
7 
8 float flag;
9 int main (int argc, char **argv) {
10 // CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num(
11 #pragma omp parallel
12 {
13 #pragma omp cancel parallel if(flag)
14   argv[0][0] = argc;
15 }
16 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
17 #pragma omp sections
18 {
19 #pragma omp cancel sections
20 }
21 // CHECK: call void @__kmpc_for_static_init_4(
22 // CHECK: call i32 @__kmpc_cancel(
23 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
24 // CHECK: call void @__kmpc_for_static_fini(
25 // CHECK: call void @__kmpc_barrier(%ident_t*
26 #pragma omp sections
27 {
28 #pragma omp cancel sections
29 #pragma omp section
30   {
31 #pragma omp cancel sections
32   }
33 }
34 // CHECK: call void @__kmpc_for_static_init_4(
35 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 3)
36 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
37 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
38 // CHECK: [[EXIT]]
39 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
40 // CHECK: br label
41 // CHECK: [[CONTINUE]]
42 // CHECK: br label
43 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 3)
44 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
45 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
46 // CHECK: [[EXIT]]
47 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
48 // CHECK: br label
49 // CHECK: [[CONTINUE]]
50 // CHECK: br label
51 // CHECK: call void @__kmpc_for_static_fini(
52 #pragma omp for
53 for (int i = 0; i < argc; ++i) {
54 #pragma omp cancel for if(cancel: flag)
55 }
56 // CHECK: call void @__kmpc_for_static_init_4(
57 // CHECK: [[FLAG:%.+]] = load float, float* @{{.+}},
58 // CHECK: [[BOOL:%.+]] = fcmp une float [[FLAG]], 0.000000e+00
59 // CHECK: br i1 [[BOOL]], label %[[THEN:[^,]+]], label %[[ELSE:[^,]+]]
60 // CHECK: [[THEN]]
61 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 2)
62 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
63 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
64 // CHECK: [[EXIT]]
65 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
66 // CHECK: br label
67 // CHECK: [[CONTINUE]]
68 // CHECK: br label
69 // CHECK: [[ELSE]]
70 // CHECK: br label
71 // CHECK: call void @__kmpc_for_static_fini(
72 // CHECK: call void @__kmpc_barrier(%ident_t*
73 #pragma omp task
74 {
75 #pragma omp cancel taskgroup
76 }
77 // CHECK: call i8* @__kmpc_omp_task_alloc(
78 // CHECK: call i32 @__kmpc_omp_task(
79 #pragma omp parallel sections
80 {
81 #pragma omp cancel sections
82 }
83 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
84 #pragma omp parallel sections
85 {
86 #pragma omp cancel sections
87 #pragma omp section
88   {
89 #pragma omp cancel sections
90   }
91 }
92 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
93 int r = 0;
94 #pragma omp parallel for reduction(+: r)
95 for (int i = 0; i < argc; ++i) {
96 #pragma omp cancel for
97   r += i;
98 }
99 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
100   return argc;
101 }
102 
103 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}},
104 // CHECK: [[FLAG:%.+]] = load float, float* @{{.+}},
105 // CHECK: [[BOOL:%.+]] = fcmp une float [[FLAG]], 0.000000e+00
106 // CHECK: br i1 [[BOOL]], label %[[THEN:[^,]+]], label %[[ELSE:[^,]+]]
107 // CHECK: [[THEN]]
108 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 {{[^,]+}}, i32 1)
109 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
110 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,]+]],
111 // CHECK: [[EXIT]]
112 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
113 // CHECK: br label %[[RETURN:.+]]
114 // CHECK: [[ELSE]]
115 // CHECK: br label
116 // CHECK: [[RETURN]]
117 // CHECK: ret void
118 
119 // CHECK: define internal i32 @{{[^(]+}}(i32
120 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 {{[^,]+}}, i32 4)
121 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
122 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,]+]],
123 // CHECK: [[EXIT]]
124 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
125 // CHECK: br label %[[RETURN:.+]]
126 // CHECK: [[RETURN]]
127 // CHECK: ret i32 0
128 
129 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}})
130 // CHECK: call void @__kmpc_for_static_init_4(
131 // CHECK: call i32 @__kmpc_cancel(
132 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
133 // CHECK: call void @__kmpc_for_static_fini(
134 // CHECK: ret void
135 
136 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}})
137 // CHECK: call void @__kmpc_for_static_init_4(
138 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID:%.+]], i32 3)
139 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
140 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
141 // CHECK: [[EXIT]]
142 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
143 // CHECK: br label
144 // CHECK: [[CONTINUE]]
145 // CHECK: br label
146 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 3)
147 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
148 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
149 // CHECK: [[EXIT]]
150 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
151 // CHECK: br label
152 // CHECK: [[CONTINUE]]
153 // CHECK: br label
154 // CHECK: call void @__kmpc_for_static_fini(
155 // CHECK: ret void
156 
157 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}},
158 // CHECK: call void @__kmpc_for_static_init_4(
159 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID:%.+]], i32 2)
160 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
161 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
162 // CHECK: [[EXIT]]
163 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
164 // CHECK: br label
165 // CHECK: [[CONTINUE]]
166 // CHECK: br label
167 // CHECK: call void @__kmpc_for_static_fini(
168 // CHECK: call i32 @__kmpc_reduce_nowait(
169 // CHECK: call void @__kmpc_end_reduce_nowait(
170 // CHECK: call void @__kmpc_for_static_fini(
171 // CHECK: ret void
172 
173 #endif
174