1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature
2; RUN: opt -S -passes='openmp-opt' < %s | FileCheck %s
3; RUN: opt -openmp-opt-disable-folding -S -passes='openmp-opt' < %s | FileCheck %s --check-prefix=CHECK-DISABLED
4; ModuleID = 'single_threaded_exeuction.c'
5
6%struct.ident_t = type { i32, i32, i32, i32, i8* }
7
8@0 = private unnamed_addr constant [1 x i8] c"\00", align 1
9@1 = private unnamed_addr constant %struct.ident_t { i32 0, i32 2, i32 0, i32 0, i8* getelementptr inbounds ([1 x i8], [1 x i8]* @0, i32 0, i32 0) }, align 8
10
11define void @kernel() {
12; CHECK-LABEL: define {{[^@]+}}@kernel() {
13; CHECK-NEXT:    [[CALL:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* nonnull @[[GLOB1:[0-9]+]], i8 1, i1 false, i1 false)
14; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[CALL]], -1
15; CHECK-NEXT:    br i1 [[CMP]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
16; CHECK:       if.then:
17; CHECK-NEXT:    call void @foo()
18; CHECK-NEXT:    br label [[IF_END:%.*]]
19; CHECK:       if.else:
20; CHECK-NEXT:    call void @bar()
21; CHECK-NEXT:    br label [[IF_END]]
22; CHECK:       if.end:
23; CHECK-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 true)
24; CHECK-NEXT:    ret void
25;
26; CHECK-DISABLED-LABEL: define {{[^@]+}}@kernel() {
27; CHECK-DISABLED-NEXT:    [[CALL:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* nonnull @[[GLOB1:[0-9]+]], i8 1, i1 false, i1 false)
28; CHECK-DISABLED-NEXT:    [[CMP:%.*]] = icmp eq i32 [[CALL]], -1
29; CHECK-DISABLED-NEXT:    br i1 [[CMP]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
30; CHECK-DISABLED:       if.then:
31; CHECK-DISABLED-NEXT:    call void @foo()
32; CHECK-DISABLED-NEXT:    br label [[IF_END:%.*]]
33; CHECK-DISABLED:       if.else:
34; CHECK-DISABLED-NEXT:    call void @bar()
35; CHECK-DISABLED-NEXT:    br label [[IF_END]]
36; CHECK-DISABLED:       if.end:
37; CHECK-DISABLED-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 true)
38; CHECK-DISABLED-NEXT:    ret void
39;
40  %call = call i32 @__kmpc_target_init(%struct.ident_t* nonnull @1, i8 1, i1 false, i1 false)
41  %cmp = icmp eq i32 %call, -1
42  br i1 %cmp, label %if.then, label %if.else
43if.then:
44  call void @foo()
45  br label %if.end
46if.else:
47  call void @bar()
48  br label %if.end
49if.end:
50  call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 true)
51  ret void
52}
53
54; Function Attrs: noinline
55define internal void @foo() {
56; CHECK-LABEL: define {{[^@]+}}@foo() {
57; CHECK-NEXT:  entry:
58; CHECK-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
59; CHECK-NEXT:    br label [[IF_THEN:%.*]]
60; CHECK:       if.then:
61; CHECK-NEXT:    call void @baz()
62; CHECK-NEXT:    br label [[IF_END:%.*]]
63; CHECK:       if.end:
64; CHECK-NEXT:    ret void
65;
66; CHECK-DISABLED-LABEL: define {{[^@]+}}@foo() {
67; CHECK-DISABLED-NEXT:  entry:
68; CHECK-DISABLED-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
69; CHECK-DISABLED-NEXT:    [[ISMAIN:%.*]] = call signext i8 @__kmpc_is_generic_main_thread_id(i32 [[TID]])
70; CHECK-DISABLED-NEXT:    [[PRED:%.*]] = icmp eq i8 [[ISMAIN]], 1
71; CHECK-DISABLED-NEXT:    br i1 [[PRED]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
72; CHECK-DISABLED:       if.then:
73; CHECK-DISABLED-NEXT:    call void @baz()
74; CHECK-DISABLED-NEXT:    br label [[IF_END]]
75; CHECK-DISABLED:       if.end:
76; CHECK-DISABLED-NEXT:    ret void
77;
78entry:
79  %tid = call i32 @__kmpc_get_hardware_thread_id();
80  %ismain = call signext i8 @__kmpc_is_generic_main_thread_id(i32 %tid)
81  %pred = icmp eq i8 %ismain, 1
82  br i1 %pred, label %if.then, label %if.end
83
84if.then:
85  call void @baz()
86  br label %if.end
87
88if.end:
89  ret void
90}
91
92; Function Attrs: noinline
93define internal void @bar() {
94; CHECK-LABEL: define {{[^@]+}}@bar() {
95; CHECK-NEXT:  entry:
96; CHECK-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
97; CHECK-NEXT:    [[ISMAIN:%.*]] = call signext i8 @__kmpc_is_generic_main_thread_id(i32 [[TID]])
98; CHECK-NEXT:    [[PRED:%.*]] = icmp eq i8 [[ISMAIN]], 1
99; CHECK-NEXT:    br i1 [[PRED]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
100; CHECK:       if.then:
101; CHECK-NEXT:    call void @baz()
102; CHECK-NEXT:    br label [[IF_END]]
103; CHECK:       if.end:
104; CHECK-NEXT:    ret void
105;
106; CHECK-DISABLED-LABEL: define {{[^@]+}}@bar() {
107; CHECK-DISABLED-NEXT:  entry:
108; CHECK-DISABLED-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
109; CHECK-DISABLED-NEXT:    [[ISMAIN:%.*]] = call signext i8 @__kmpc_is_generic_main_thread_id(i32 [[TID]])
110; CHECK-DISABLED-NEXT:    [[PRED:%.*]] = icmp eq i8 [[ISMAIN]], 1
111; CHECK-DISABLED-NEXT:    br i1 [[PRED]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
112; CHECK-DISABLED:       if.then:
113; CHECK-DISABLED-NEXT:    call void @baz()
114; CHECK-DISABLED-NEXT:    br label [[IF_END]]
115; CHECK-DISABLED:       if.end:
116; CHECK-DISABLED-NEXT:    ret void
117;
118entry:
119  %tid = call i32 @__kmpc_get_hardware_thread_id();
120  %ismain = call signext i8 @__kmpc_is_generic_main_thread_id(i32 %tid)
121  %pred = icmp eq i8 %ismain, 1
122  br i1 %pred, label %if.then, label %if.end
123
124if.then:
125  call void @baz()
126  br label %if.end
127
128if.end:
129  ret void
130}
131
132declare void @baz()
133
134declare i8 @__kmpc_is_generic_main_thread_id(i32)
135
136declare i32 @__kmpc_get_hardware_thread_id()
137
138declare i32 @__kmpc_target_init(%struct.ident_t*, i8, i1, i1)
139
140declare void @__kmpc_target_deinit(%struct.ident_t*, i8, i1)
141
142!llvm.dbg.cu = !{!0}
143!llvm.module.flags = !{!3, !4, !5, !6}
144!nvvm.annotations = !{!7}
145
146!0 = distinct !DICompileUnit(language: DW_LANG_C99, file: !1, producer: "clang version 13.0.0", isOptimized: false, runtimeVersion: 0, emissionKind: FullDebug, enums: !2, splitDebugInlining: false, nameTableKind: None)
147!1 = !DIFile(filename: "fold_generic_main_thread.c", directory: "/tmp/fold_generic_main_thread.c")
148!2 = !{}
149!3 = !{i32 2, !"Debug Info Version", i32 3}
150!4 = !{i32 1, !"wchar_size", i32 4}
151!5 = !{i32 7, !"openmp", i32 50}
152!6 = !{i32 7, !"openmp-device", i32 50}
153!7 = !{void ()* @kernel, !"kernel", i32 1}
154