Florian Hahn 8bc038daf2
[InstComb] Allow more user for (add (ptrtoint %B), %O) to GEP transform. (#153566)
Generalize the logic from
https://github.com/llvm/llvm-project/pull/153421 to support additional
cases where the pointer is only used as integer.

Alive2 Proof: https://alive2.llvm.org/ce/z/po58pP

This enables vectorizing std::find for some cases, if additional
assumptions are provided: https://godbolt.org/z/94oq3576E

Depends on https://github.com/llvm/llvm-project/pull/15342.

PR: https://github.com/llvm/llvm-project/pull/153566
2025-08-22 10:17:12 +01:00

70 lines
2.1 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -passes=instcombine -S %s | FileCheck %s
define i64 @inttoptr_used_by_phi_with_ptrtoint(i1 %c, ptr %src, ptr %p2) {
; CHECK-LABEL: define i64 @inttoptr_used_by_phi_with_ptrtoint(
; CHECK-SAME: i1 [[C:%.*]], ptr [[SRC:%.*]], ptr [[P2:%.*]]) {
; CHECK-NEXT: br i1 [[C]], label %[[THEN:.*]], label %[[ELSE:.*]]
; CHECK: [[THEN]]:
; CHECK-NEXT: [[P:%.*]] = getelementptr i8, ptr [[SRC]], i64 10
; CHECK-NEXT: [[TMP1:%.*]] = ptrtoint ptr [[P]] to i64
; CHECK-NEXT: br label %[[EXIT:.*]]
; CHECK: [[ELSE]]:
; CHECK-NEXT: br label %[[EXIT]]
; CHECK: [[EXIT]]:
; CHECK-NEXT: [[PHI:%.*]] = phi i64 [ [[TMP1]], %[[THEN]] ], [ 0, %[[ELSE]] ]
; CHECK-NEXT: ret i64 [[PHI]]
;
%i = ptrtoint ptr %src to i64
%a = add i64 %i, 10
%p = inttoptr i64 %a to ptr
br i1 %c, label %then, label %else
then:
br label %exit
else:
br label %exit
exit:
%phi = phi ptr [ %p, %then ], [ null, %else ]
%i.2 = ptrtoint ptr %phi to i64
ret i64 %i.2
}
declare void @foo(ptr)
define i64 @inttoptr_used_by_phi_with_other_users(i1 %c, ptr %src, ptr %p2) {
; CHECK-LABEL: define i64 @inttoptr_used_by_phi_with_other_users(
; CHECK-SAME: i1 [[C:%.*]], ptr [[SRC:%.*]], ptr [[P2:%.*]]) {
; CHECK-NEXT: [[I:%.*]] = ptrtoint ptr [[SRC]] to i64
; CHECK-NEXT: [[A:%.*]] = add i64 [[I]], 10
; CHECK-NEXT: br i1 [[C]], label %[[THEN:.*]], label %[[ELSE:.*]]
; CHECK: [[THEN]]:
; CHECK-NEXT: br label %[[EXIT:.*]]
; CHECK: [[ELSE]]:
; CHECK-NEXT: br label %[[EXIT]]
; CHECK: [[EXIT]]:
; CHECK-NEXT: [[PHI:%.*]] = phi i64 [ [[A]], %[[THEN]] ], [ 0, %[[ELSE]] ]
; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[A]] to ptr
; CHECK-NEXT: call void @foo(ptr [[P]])
; CHECK-NEXT: ret i64 [[PHI]]
;
%i = ptrtoint ptr %src to i64
%a = add i64 %i, 10
%p = inttoptr i64 %a to ptr
br i1 %c, label %then, label %else
then:
br label %exit
else:
br label %exit
exit:
%phi = phi ptr [ %p, %then ], [ null, %else ]
call void @foo(ptr %p)
%i.2 = ptrtoint ptr %phi to i64
ret i64 %i.2
}