Lei Wang b8cc3ba409
[PseudoProbe] Extend to skip instrumenting probe into the dests of invoke (#79919)
As before we only skip instrumenting probe of `unwind`(`KnownColdBlock`)
block, this PR extends to skip the both EH flow from `invoke`, i.e. also
skip the `normal` dest. For more contexts: when doing call-to-invoke
conversion, the block is split by the `invoke` and two extra
blocks(`normal` and `unwind`) are added. With this PR, the
instrumentation is the same as the one before the call-to-invoke
conversion.
One significant benefit is this can help mitigate the "unstable IR"
issue(https://discourse.llvm.org/t/ipo-for-linkonce-odr-functions/69404),
the two versions now are on the same probe instrumentation, expected to
be the same checksum.
To achieve the same checksum, some tweaks is needed:
-  Now it also skips incrementing the probe ID for the skipped probe.
- The checksum is also computed based on the CFG that skips the EH
edges.

We observed this fixes ~5% mismatched samples.
2024-04-01 13:54:54 -07:00

44 lines
1.2 KiB
LLVM

; REQUIRES: x86_64-linux
; RUN: opt < %s -passes=pseudo-probe -function-sections -S -o - | FileCheck %s
;; Check the generation of pseudoprobe intrinsic call for non-EH blocks only.
declare i32 @__gxx_personality_v0(...)
declare i32 @llvm.eh.typeid.for(ptr) nounwind
declare ptr @__cxa_begin_catch(ptr)
declare void @__cxa_end_catch()
declare void @bar()
@_ZTIi = external constant ptr
define void @foo() uwtable ssp personality ptr @__gxx_personality_v0 {
entry:
; CHECK: call void @llvm.pseudoprobe
invoke void @bar()
to label %ret unwind label %lpad
ret:
; CHECK-NOT: call void @llvm.pseudoprobe
ret void
lpad: ; preds = %entry
; CHECK-NOT: call void @llvm.pseudoprobe
%exn = landingpad {ptr, i32}
catch ptr @_ZTIi
%eh.exc = extractvalue { ptr, i32 } %exn, 0
%eh.selector = extractvalue { ptr, i32 } %exn, 1
%0 = call i32 @llvm.eh.typeid.for(ptr @_ZTIi) nounwind
%1 = icmp eq i32 %eh.selector, %0
br i1 %1, label %catch, label %eh.resume
catch:
; CHECK-NOT: call void @llvm.pseudoprobe
%ignored = call ptr @__cxa_begin_catch(ptr %eh.exc) nounwind
call void @__cxa_end_catch() nounwind
br label %ret
eh.resume:
; CHECK-NOT: call void @llvm.pseudoprobe
resume { ptr, i32 } %exn
}