[rtsan] Add fork/execve interceptors (#117198)
[llvm-project.git] / llvm / test / Transforms / PGOProfile / icp_vtable_invoke.ll
blob6d3a6972f6885cfe14b453ff94bbe6cc9f977804
1 ; RUN: opt < %s -passes='pgo-icall-prom' -enable-vtable-profile-use -S | FileCheck %s --check-prefix=VTABLE
3 target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128-f80:128-n8:16:32:64-S128"
4 target triple = "x86_64-unknown-linux-gnu"
6 @_ZTV4Base = constant { [3 x ptr] } { [3 x ptr] [ptr null, ptr null, ptr @_ZN4Base10get_ticketEv] }, !type !0, !type !1
7 @_ZTV7Derived = constant { [3 x ptr] } { [3 x ptr] [ptr null, ptr null, ptr @_ZN7Derived10get_ticketEv] }, !type !0, !type !1, !type !2, !type !3
9 @.str = private constant [15 x i8] c"out of tickets\00"
11 define i32 @test(ptr %b) personality ptr @__gxx_personality_v0 {
12 ; VTABLE-LABEL: define i32 @test(
13 ; VTABLE-SAME: ptr [[B:%.*]]) personality ptr @__gxx_personality_v0 {
14 ; VTABLE-NEXT:  [[ENTRY:.*:]]
15 ; VTABLE-NEXT:    [[VTABLE:%.*]] = load ptr, ptr [[B]], align 8
16 ; VTABLE-NEXT:    [[TMP0:%.*]] = tail call i1 @llvm.type.test(ptr [[VTABLE]], metadata !"_ZTS4Base")
17 ; VTABLE-NEXT:    tail call void @llvm.assume(i1 [[TMP0]])
18 ; VTABLE-NEXT:    [[TMP3:%.*]] = icmp eq ptr [[VTABLE]], getelementptr inbounds (i8, ptr @_ZTV7Derived, i32 16)
19 ; VTABLE-NEXT:    br i1 [[TMP3]], label %[[IF_TRUE_DIRECT_TARG:.*]], label %[[IF_FALSE_ORIG_INDIRECT:.*]], !prof [[PROF4:![0-9]+]]
20 ; VTABLE:       [[IF_TRUE_DIRECT_TARG]]:
21 ; VTABLE-NEXT:    [[TMP2:%.*]] = invoke i32 @_ZN7Derived10get_ticketEv(ptr [[B]])
22 ; VTABLE-NEXT:            to label %[[IF_END_ICP:.*]] unwind label %[[LPAD:.*]]
23 ; VTABLE:       [[IF_FALSE_ORIG_INDIRECT]]:
24 ; VTABLE-NEXT:    [[TMP4:%.*]] = icmp eq ptr [[VTABLE]], getelementptr inbounds (i8, ptr @_ZTV4Base, i32 16)
25 ; VTABLE-NEXT:    br i1 [[TMP4]], label %[[IF_TRUE_DIRECT_TARG1:.*]], label %[[IF_FALSE_ORIG_INDIRECT2:.*]], !prof [[PROF5:![0-9]+]]
26 ; VTABLE:       [[IF_TRUE_DIRECT_TARG1]]:
27 ; VTABLE-NEXT:    [[TMP5:%.*]] = invoke i32 @_ZN4Base10get_ticketEv(ptr [[B]])
28 ; VTABLE-NEXT:            to label %[[IF_END_ICP3:.*]] unwind label %[[LPAD]]
29 ; VTABLE:       [[IF_FALSE_ORIG_INDIRECT2]]:
30 ; VTABLE-NEXT:    [[TMP1:%.*]] = load ptr, ptr [[VTABLE]], align 8
31 ; VTABLE-NEXT:    [[CALL:%.*]] = invoke i32 [[TMP1]](ptr [[B]])
32 ; VTABLE-NEXT:            to label %[[IF_END_ICP3]] unwind label %[[LPAD]]
33 ; VTABLE:       [[IF_END_ICP3]]:
34 ; VTABLE-NEXT:    [[TMP6:%.*]] = phi i32 [ [[CALL]], %[[IF_FALSE_ORIG_INDIRECT2]] ], [ [[TMP5]], %[[IF_TRUE_DIRECT_TARG1]] ]
35 ; VTABLE-NEXT:    br label %[[IF_END_ICP]]
36 ; VTABLE:       [[IF_END_ICP]]:
37 ; VTABLE-NEXT:    [[TMP7:%.*]] = phi i32 [ [[TMP6]], %[[IF_END_ICP3]] ], [ [[TMP2]], %[[IF_TRUE_DIRECT_TARG]] ]
38 ; VTABLE-NEXT:    br label %[[NEXT:.*]]
39 ; VTABLE:       [[NEXT]]:
40 ; VTABLE-NEXT:    ret i32 [[TMP7]]
41 ; VTABLE:       [[LPAD]]:
42 ; VTABLE-NEXT:    [[EXN:%.*]] = landingpad { ptr, i32 }
43 ; VTABLE-NEXT:            cleanup
44 ; VTABLE-NEXT:    unreachable
46 entry:
47   %vtable = load ptr, ptr %b, !prof !4
48   %0 = tail call i1 @llvm.type.test(ptr %vtable, metadata !"_ZTS4Base")
49   tail call void @llvm.assume(i1 %0)
50   %1 = load ptr, ptr %vtable
51   %call = invoke i32 %1(ptr %b) to label %next unwind label %lpad, !prof !5
53 next:
54   ret i32 %call
56 lpad:
57   %exn = landingpad {ptr, i32}
58   cleanup
59   unreachable
62 declare void @make_error(ptr, ptr, i32)
63 declare i32 @get_ticket_id()
64 declare ptr @__cxa_allocate_exception(i64)
66 define i32 @_ZN4Base10get_ticketEv(ptr %this) personality ptr @__gxx_personality_v0 {
67 entry:
68   %call = tail call i32 @get_ticket_id()
69   %cmp.not = icmp eq i32 %call, -1
70   br i1 %cmp.not, label %if.end, label %if.then
72 if.then:
73   ret i32 %call
75 if.end:
76   %exception = tail call ptr @__cxa_allocate_exception(i64 1)
77   invoke void @make_error(ptr %exception, ptr @.str, i32 1)
78   to label %invoke.cont unwind label %lpad
80 invoke.cont:
81   unreachable
83 lpad:
84   %0 = landingpad { ptr, i32 }
85   cleanup
86   resume { ptr, i32 } %0
89 define i32 @_ZN7Derived10get_ticketEv(ptr %this) personality ptr @__gxx_personality_v0 {
90 entry:
91   %call = tail call i32 @get_ticket_id()
92   %cmp.not = icmp eq i32 %call, -1
93   br i1 %cmp.not, label %if.end, label %if.then
95 if.then:
96   ret i32 %call
98 if.end:
99   %exception = tail call ptr @__cxa_allocate_exception(i64 1)
100   invoke void @make_error(ptr %exception, ptr @.str, i32 2)
101   to label %invoke.cont unwind label %lpad
103 invoke.cont:
104   unreachable
106 lpad:
107   %0 = landingpad { ptr, i32 }
108   cleanup
109   resume { ptr, i32 } %0
112 declare i1 @llvm.type.test(ptr, metadata)
113 declare void @llvm.assume(i1)
114 declare i32 @__gxx_personality_v0(...)
116 !0 = !{i64 16, !"_ZTS4Base"}
117 !1 = !{i64 16, !"_ZTSM4BaseFivE.virtual"}
118 !2 = !{i64 16, !"_ZTS7Derived"}
119 !3 = !{i64 16, !"_ZTSM7DerivedFivE.virtual"}
120 !4 = !{!"VP", i32 2, i64 1600, i64 13870436605473471591, i64 900, i64 1960855528937986108, i64 700}
121 !5 = !{!"VP", i32 0, i64 1600, i64 14811317294552474744, i64 900, i64 9261744921105590125, i64 700}
123 ; VTABLE: [[PROF4]] = !{!"branch_weights", i32 900, i32 700}
124 ; VTABLE: [[PROF5]] = !{!"branch_weights", i32 700, i32 0}