llvm-project/llvm/test/Transforms/PhaseOrdering/phi-protected-field-ptr.ll
Peter Collingbourne 75bb30ddbf
Move {load,store}(llvm.protected.field.ptr) lowering to InstCombine.
The previous position of llvm.protected.field.ptr lowering for loads
and stores was problematic as it not only inhibited optimizations such
as DSE (as stores to a llvm.protected.field.ptr were not considered to
must-alias stores to the non-protected.field pointer) but also required
changes to other optimization passes to avoid transformations that would
reduce PFP coverage.

Address this by moving the load/store part of the lowering to
InstCombine, where it will run earlier than the PFP-breaking and
AA-relying transformations. The deactivation symbol, null comparison
and EmuPAC parts of the lowering remain in PreISelLowering.

Now that the transformation inhibitions are no longer needed, remove them
(i.e. partially revert #151649, and revert #182976).

This change resulted in a 2.4% reduction in Fleetbench .text size and
the following improvements to PFP performance overhead for BM_PROTO_Arena
on various microarchitectures:

                    before   after
  Apple M2 Ultra     3.5%    3.3%
  Google Axion C4A   3.3%    2.9%
  Google Axion N4A   2.7%    2.2%

Reviewers: fmayer, nikic, vitalybuka

Reviewed By: fmayer

Pull Request: https://github.com/llvm/llvm-project/pull/186548
2026-04-06 17:47:24 -07:00

43 lines
1.7 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -O2 -S < %s | FileCheck %s
; Test that no optimization run at -O2 before InstCombine moves the loads into
; the exit block, as this causes unnecessary address escapes with pointer field
; protection.
define ptr @phi_prot_ptr(i1 %sel, ptr %p1, ptr %p2) {
; CHECK-LABEL: define ptr @phi_prot_ptr(
; CHECK-SAME: i1 [[SEL:%.*]], ptr readonly captures(none) [[P1:%.*]], ptr readonly captures(none) [[P2:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
; CHECK-NEXT: br i1 [[SEL]], label %[[T:.*]], label %[[F:.*]]
; CHECK: [[T]]:
; CHECK-NEXT: [[TMP1:%.*]] = load ptr, ptr [[P1]], align 8
; CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
; CHECK-NEXT: [[TMP3:%.*]] = tail call i64 @llvm.ptrauth.auth(i64 [[TMP2]], i32 2, i64 1)
; CHECK-NEXT: br label %[[EXIT:.*]]
; CHECK: [[F]]:
; CHECK-NEXT: [[TMP4:%.*]] = load ptr, ptr [[P2]], align 8
; CHECK-NEXT: [[TMP5:%.*]] = ptrtoint ptr [[TMP4]] to i64
; CHECK-NEXT: [[TMP6:%.*]] = tail call i64 @llvm.ptrauth.auth(i64 [[TMP5]], i32 2, i64 2)
; CHECK-NEXT: br label %[[EXIT]]
; CHECK: [[EXIT]]:
; CHECK-NEXT: [[RETVAL_IN:%.*]] = phi i64 [ [[TMP3]], %[[T]] ], [ [[TMP6]], %[[F]] ]
; CHECK-NEXT: [[RETVAL:%.*]] = inttoptr i64 [[RETVAL_IN]] to ptr
; CHECK-NEXT: ret ptr [[RETVAL]]
;
br i1 %sel, label %t, label %f
t:
%protp1 = call ptr @llvm.protected.field.ptr.p0(ptr %p1, i64 1, i1 true)
%load1 = load ptr, ptr %protp1
br label %exit
f:
%protp2 = call ptr @llvm.protected.field.ptr.p0(ptr %p2, i64 2, i1 true)
%load2 = load ptr, ptr %protp2
br label %exit
exit:
%retval = phi ptr [ %load1, %t ], [ %load2, %f ]
ret ptr %retval
}