xref: /llvm-project/clang/test/OpenMP/masked_codegen.cpp (revision 0c6f2f629cc0017361310fa4c132090413a874db)
1 // RUN: %clang_cc1 -verify -fopenmp -x c++ -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s --check-prefixes=ALL,NORMAL
2 // RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
3 // RUN: %clang_cc1 -fopenmp -x c++ -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefixes=ALL,NORMAL
4 // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp -fexceptions -fcxx-exceptions -debug-info-kind=line-tables-only -x c++ -emit-llvm %s -o - | FileCheck %s --check-prefix=TERM_DEBUG
5 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-enable-irbuilder -x c++ -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s --check-prefixes=ALL,IRBUILDER
6 // RUN: %clang_cc1 -fopenmp -fopenmp-enable-irbuilder -x c++ -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
7 // RUN: %clang_cc1 -fopenmp -fopenmp-enable-irbuilder -x c++ -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefixes=ALL,IRBUILDER
8 
9 // RUN: %clang_cc1 -verify -fopenmp-simd -x c++ -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck --check-prefix SIMD-ONLY0 %s
10 // RUN: %clang_cc1 -fopenmp-simd -x c++ -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
11 // RUN: %clang_cc1 -fopenmp-simd -x c++ -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
12 // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp-simd -fexceptions -fcxx-exceptions -debug-info-kind=line-tables-only -x c++ -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
13 // SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
14 // expected-no-diagnostics
15 #ifndef HEADER
16 #define HEADER
17 
18 // ALL:       [[IDENT_T_TY:%.+]] = type { i32, i32, i32, i32, ptr }
19 
20 // ALL:       define {{.*}}void [[FOO:@.+]]()
21 
foo()22 void foo() { extern void mayThrow(); mayThrow(); }
23 
24 // ALL-LABEL: @main
25 // TERM_DEBUG-LABEL: @main
main()26 int main() {
27   // ALL:      			[[A_ADDR:%.+]] = alloca i8
28   char a;
29 
30 // ALL:       			[[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
31 // ALL:       			[[RES:%.+]] = call {{.*}}i32 @__kmpc_masked(ptr [[DEFAULT_LOC]], i32 [[GTID]], i32 0)
32 // ALL-NEXT:  			[[IS_MASKED:%.+]] = icmp ne i32 [[RES]], 0
33 // ALL-NEXT:  			br i1 [[IS_MASKED]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
34 // ALL:       			[[THEN]]
35 // ALL-NEXT:  			store i8 2, ptr [[A_ADDR]]
36 // IRBUILDER-NEXT:		br label %[[AFTER:[^ ,]+]]
37 // IRBUILDER:			[[AFTER]]
38 // ALL-NEXT:  			call {{.*}}void @__kmpc_end_masked(ptr [[DEFAULT_LOC]], i32 [[GTID]])
39 // ALL-NEXT:  			br label {{%?}}[[EXIT]]
40 // ALL:       			[[EXIT]]
41 #pragma omp masked
42   a = 2;
43 // IRBUILDER: 			[[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
44 // ALL:       			[[RES:%.+]] = call {{.*}}i32 @__kmpc_masked(ptr [[DEFAULT_LOC]], i32 [[GTID]], i32 2)
45 // ALL-NEXT:  			[[IS_MASKED:%.+]] = icmp ne i32 [[RES]], 0
46 // ALL-NEXT:  			br i1 [[IS_MASKED]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
47 // ALL:       			[[THEN]]
48 // IRBUILDER-NEXT:  call {{.*}}void [[FOO]]()
49 // NORMAL-NEXT:  		invoke {{.*}}void [[FOO]]()
50 // ALL:       			call {{.*}}void @__kmpc_end_masked(ptr [[DEFAULT_LOC]], i32 [[GTID]])
51 // ALL-NEXT:  			br label {{%?}}[[EXIT]]
52 // ALL:       			[[EXIT]]
53 #pragma omp masked filter(2)
54   foo();
55 // ALL:                         store i32 9, ptr [[X:.+]],
56 // ALL:                         [[X_VAL:%.+]] = load i32, ptr [[X]]
57 // IRBUILDER: 			[[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
58 // ALL:       			[[RES:%.+]] = call {{.*}}i32 @__kmpc_masked(ptr [[DEFAULT_LOC]], i32 [[GTID]], i32 [[X_VAL]])
59 // ALL-NEXT:  			[[IS_MASKED:%.+]] = icmp ne i32 [[RES]], 0
60 // ALL-NEXT:  			br i1 [[IS_MASKED]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
61 // ALL:       			[[THEN]]
62 // IRBUILDER-NEXT:  call {{.*}}void [[FOO]]()
63 // NORMAL-NEXT:  		invoke {{.*}}void [[FOO]]()
64 // ALL:       			call {{.*}}void @__kmpc_end_masked(ptr [[DEFAULT_LOC]], i32 [[GTID]])
65 // ALL-NEXT:  			br label {{%?}}[[EXIT]]
66 // ALL:       			[[EXIT]]
67   int x = 9;
68 #pragma omp masked filter(x)
69   foo();
70   // ALL-NOT:   call i32 @__kmpc_masked
71   // ALL-NOT:   call void @__kmpc_end_masked
72   return a;
73 }
74 
75 // ALL-LABEL:        lambda_masked
76 // TERM_DEBUG-LABEL: lambda_masked
lambda_masked(int a,int b)77 void lambda_masked(int a, int b) {
78   auto l = [=]() {
79 #pragma omp masked
80     {
81       // ALL: call i32 @__kmpc_masked(
82       int c = a + b;
83     }
84   };
85 
86   l();
87 
88   auto l1 = [=]() {
89 #pragma omp parallel
90 #pragma omp masked filter(1)
91     {
92       // ALL: call i32 @__kmpc_masked(
93       int c = a + b;
94     }
95   };
96 
97   l1();
98 
99   int y = 1;
100   auto l2 = [=](int yy) {
101 #pragma omp parallel
102 #pragma omp masked filter(yy)
103     {
104       // ALL: call i32 @__kmpc_masked(
105       int c = a + b;
106     }
107   };
108 
109   l2(y);
110 }
111 
112 // ALL-LABEL:      parallel_masked
113 // TERM_DEBUG-LABEL: parallel_masked
parallel_masked()114 void parallel_masked() {
115 #pragma omp parallel
116 #pragma omp masked filter(1)
117   // TERM_DEBUG-NOT: __kmpc_global_thread_num
118   // TERM_DEBUG:     call i32 @__kmpc_masked({{.+}}), !dbg [[DBG_LOC_START:![0-9]+]]
119   // TERM_DEBUG:     invoke void {{.*}}foo{{.*}}()
120   // TERM_DEBUG:     unwind label %[[TERM_LPAD:.+]],
121   // TERM_DEBUG-NOT: __kmpc_global_thread_num
122   // TERM_DEBUG:     call void @__kmpc_end_masked({{.+}}), !dbg [[DBG_LOC_END:![0-9]+]]
123   // TERM_DEBUG:     [[TERM_LPAD]]
124   // TERM_DEBUG:     call void @__clang_call_terminate
125   // TERM_DEBUG:     unreachable
126   foo();
127 
128   int x;
129 #pragma omp parallel
130 #pragma omp masked filter(x)
131   // TERM_DEBUG-NOT: __kmpc_global_thread_num
132   // TERM_DEBUG:     call i32 @__kmpc_masked({{.+}}), !dbg [[DBG_LOC_START:![0-9]+]]
133   // TERM_DEBUG:     invoke void {{.*}}foo{{.*}}()
134   // TERM_DEBUG:     unwind label %[[TERM_LPAD:.+]],
135   // TERM_DEBUG-NOT: __kmpc_global_thread_num
136   // TERM_DEBUG:     call void @__kmpc_end_masked({{.+}}), !dbg [[DBG_LOC_END:![0-9]+]]
137   // TERM_DEBUG:     [[TERM_LPAD]]
138   // TERM_DEBUG:     call void @__clang_call_terminate
139   // TERM_DEBUG:     unreachable
140   foo();
141 }
142 // TERM_DEBUG-DAG: [[DBG_LOC_START]] = !DILocation(line: [[@LINE-12]],
143 // TERM_DEBUG-DAG: [[DBG_LOC_END]] = !DILocation(line: [[@LINE-3]],
144 
145 #endif
146