llvm-project/llvm/test/Transforms/InstCombine/non-integral-pointers.ll
Nikita Popov 90ba33099c
[InstCombine] Canonicalize constant GEPs to i8 source element type (#68882)
This patch canonicalizes getelementptr instructions with constant
indices to use the `i8` source element type. This makes it easier for
optimizations to recognize that two GEPs are identical, because they
don't need to see past many different ways to express the same offset.

This is a first step towards
https://discourse.llvm.org/t/rfc-replacing-getelementptr-with-ptradd/68699.
This is limited to constant GEPs only for now, as they have a clear
canonical form, while we're not yet sure how exactly to deal with
variable indices.

The test llvm/test/Transforms/PhaseOrdering/switch_with_geps.ll gives
two representative examples of the kind of optimization improvement we
expect from this change. In the first test SimplifyCFG can now realize
that all switch branches are actually the same. In the second test it
can convert it into simple arithmetic. These are representative of
common optimization failures we see in Rust.

Fixes https://github.com/llvm/llvm-project/issues/69841.
2024-01-24 15:25:29 +01:00

102 lines
3.5 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -passes=instcombine -S < %s | FileCheck %s
target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128-ni:4"
target triple = "x86_64-unknown-linux-gnu"
define ptr addrspace(4) @f_0() {
; CHECK-LABEL: @f_0(
; CHECK-NEXT: ret ptr addrspace(4) getelementptr (i8, ptr addrspace(4) null, i64 50)
;
%result = getelementptr i8, ptr addrspace(4) null, i64 50
ret ptr addrspace(4) %result
}
define ptr addrspace(3) @f_1() {
; inttoptr is fine here since addrspace(3) is integral.
; CHECK-LABEL: @f_1(
; CHECK-NEXT: ret ptr addrspace(3) inttoptr (i64 50 to ptr addrspace(3))
;
%result = getelementptr i8, ptr addrspace(3) null, i64 50
ret ptr addrspace(3) %result
}
define void @f_2(ptr %ptr0, ptr %ptr1) {
; It is not okay to convert the load/store pair to load and store
; integers, since pointers in address space 4 are non-integral.
; CHECK-LABEL: @f_2(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[VAL:%.*]] = load ptr addrspace(4), ptr [[PTR0:%.*]], align 8
; CHECK-NEXT: store ptr addrspace(4) [[VAL]], ptr [[PTR1:%.*]], align 8
; CHECK-NEXT: ret void
;
entry:
%val = load ptr addrspace(4), ptr %ptr0
store ptr addrspace(4) %val, ptr %ptr1
ret void
}
define void @f_3(ptr %ptr0, ptr %ptr1) {
; It *is* okay to convert the load/store pair to load and store
; integers, since pointers in address space 3 are integral.
; CHECK-LABEL: @f_3(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[VAL:%.*]] = load ptr addrspace(3), ptr [[PTR0:%.*]], align 8
; CHECK-NEXT: store ptr addrspace(3) [[VAL]], ptr [[PTR1:%.*]], align 8
; CHECK-NEXT: ret void
;
entry:
%val = load ptr addrspace(3), ptr %ptr0
store ptr addrspace(3) %val, ptr %ptr1
ret void
}
define i64 @g(ptr %gp) {
; CHECK-LABEL: @g(
; CHECK-NEXT: [[DOTPRE:%.*]] = load ptr addrspace(4), ptr [[GP:%.*]], align 8
; CHECK-NEXT: [[V74:%.*]] = call ptr addrspace(4) @alloc()
; CHECK-NEXT: [[V75:%.*]] = addrspacecast ptr addrspace(4) [[V74]] to ptr
; CHECK-NEXT: [[V77:%.*]] = getelementptr i8, ptr [[V75]], i64 -8
; CHECK-NEXT: store ptr addrspace(4) [[DOTPRE]], ptr [[V77]], align 8
; CHECK-NEXT: [[V81:%.*]] = load i64, ptr [[V77]], align 8
; CHECK-NEXT: ret i64 [[V81]]
;
%.pre = load ptr addrspace(4), ptr %gp, align 8
%v74 = call ptr addrspace(4) @alloc()
%v75 = addrspacecast ptr addrspace(4) %v74 to ptr
%v77 = getelementptr ptr addrspace(4), ptr %v75, i64 -1
store ptr addrspace(4) %.pre, ptr %v77, align 8
%v81 = load i64, ptr %v77, align 8
ret i64 %v81
}
define i64 @g2(ptr addrspace(4) %gp) {
; CHECK-LABEL: @g2(
; CHECK-NEXT: [[DOTPRE:%.*]] = load ptr, ptr addrspace(4) [[GP:%.*]], align 8
; CHECK-NEXT: [[V74:%.*]] = call ptr addrspace(4) @alloc()
; CHECK-NEXT: [[V77:%.*]] = getelementptr i8, ptr addrspace(4) [[V74]], i64 -8
; CHECK-NEXT: store ptr [[DOTPRE]], ptr addrspace(4) [[V77]], align 8
; CHECK-NEXT: [[V81_CAST:%.*]] = ptrtoint ptr [[DOTPRE]] to i64
; CHECK-NEXT: ret i64 [[V81_CAST]]
;
%.pre = load ptr, ptr addrspace(4) %gp, align 8
%v74 = call ptr addrspace(4) @alloc()
%v77 = getelementptr ptr, ptr addrspace(4) %v74, i64 -1
store ptr %.pre, ptr addrspace(4) %v77, align 8
%v81 = load i64, ptr addrspace(4) %v77, align 8
ret i64 %v81
}
declare ptr addrspace(4) @alloc()
define i64 @f_4(ptr addrspace(4) %v0) {
; CHECK-LABEL: @f_4(
; CHECK-NEXT: [[V6:%.*]] = call i64 @f_5(ptr addrspace(4) [[V0:%.*]])
; CHECK-NEXT: ret i64 [[V6]]
;
%v6 = call i64 @f_5(ptr addrspace(4) %v0)
ret i64 %v6
}
declare i64 @f_5(i64)