llvm-project/llvm/test/CodeGen/RISCV/stack-clash-prologue.ll
Alex Bradbury 3d2650bdeb
[RISCV] Use addi rather than addiw for immediates materialised by lui+addi(w) pairs when possible (#141663)
The logic in RISCVMatInt would previously produce lui+addiw on RV64
whenever a 32-bit integer must be materialised and the Hi20 and Lo12
parts are non-zero. However, sometimes addi can be used equivalently
(whenever the sign extension behaviour of addiw would be a no-op). This
patch moves to using addiw only when necessary. Although there is
absolutely no advantage in terms of compressibility or performance, this
has the following advantages:
* It's more consistent with logic used elsewhere in the backend. For
instance, RISCVOptWInstrs will try to convert addiw to addi on the basis
it reduces test diffs vs RV32.
* This matches the lowering GCC does in its codegen path. Unlike LLVM,
GCC seems to have different expansion logic for the assembler vs
codegen. For codegen it will use lui+addi if possible, but expanding
`li` in the assembler will always produces lui+addiw as LLVM did prior
to this commit. As someone who has been looking at a lot of gcc vs clang
diffs lately, reducing unnecessary divergence is of at least some value.
* As the diff for fold-mem-offset.ll shows, we can fold memory offsets
in more cases when addi is used. Memory offset folding could be taught
to recognise when the addiw could be replaced with an addi, but that
seems unnecessary when we can simply change the logic in RISCVMatInt.

As pointed out by @topperc during review, making this change without
modifying RISCVOptWInstrs risks introducing some cases where we fail to
remove a sext.w that we removed before. I've incorporated a patch based
on a suggestion from Craig that avoids it, and also adds appropriate
RISCVOptWInstrs test cases.

The initial patch description noted that the main motivation was to
avoid unnecessary differences both for RV32/RV64 and when comparing GCC,
but noted that very occasionally we see a benefit from memory offset
folding kicking in when it didn't before. Looking at the dynamic
instruction count difference for SPEC benchmarks targeting rva22u64 and
it shows we actually get a meaningful
~4.3% reduction in dynamic icount for 519.lbm_r. Looking at the data
more closely, the codegen difference is in `LBM_performStreamCollideTRT`
which as a function accounts for ~98% for dynamically executed
instructions and the codegen diffs appear to be a knock-on effect of the
address merging reducing register pressure right from function entry
(for instance, we get a big reduction in dynamically executed loads in
that function).

Below is the icount data (rva22u64 -O3, no LTO):
```
Benchmark                Baseline            This PR   Diff (%)
============================================================
500.perlbench_r         174116601991    174115795810     -0.00%
502.gcc_r               218903280858    218903215788     -0.00%
505.mcf_r               131208029185    131207692803     -0.00%
508.namd_r              217497594322    217497594297     -0.00%
510.parest_r            289314486153    289313577652     -0.00%
511.povray_r             30640531048     30640765701      0.00%
519.lbm_r                95897914862     91712688050     -4.36%
520.omnetpp_r           134641549722    134867015683      0.17%
523.xalancbmk_r         281462762992    281432092673     -0.01%
525.x264_r              379776121941    379535558210     -0.06%
526.blender_r           659736022025    659738387343      0.00%
531.deepsjeng_r         349122867552    349122867481     -0.00%
538.imagick_r           238558760552    238558753269     -0.00%
541.leela_r             406578560612    406385135260     -0.05%
544.nab_r               400997131674    400996765827     -0.00%
557.xz_r                130079522194    129945515709     -0.10%

```

The instcounting setup I use doesn't have good support for drilling down
into functions from outside the linked executable (e.g. libc). The
difference in omnetpp all seems to come from there, and does not reflect
any degradation in codegen quality.

I can confirm with the current version of the PR there is no change in
the number of static sext.w across all the SPEC 2017 benchmarks
(rva22u64 O3)

Co-authored-by: Craig Topper <craig.topper@sifive.com>
2025-06-02 22:24:50 +01:00

735 lines
23 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -mtriple=riscv64 -mattr=+m -O2 < %s \
; RUN: | FileCheck %s -check-prefix=RV64I
; RUN: llc -mtriple=riscv32 -mattr=+m -O2 < %s \
; RUN: | FileCheck %s -check-prefix=RV32I
; Tests copied from PowerPC.
; Free probe
define i8 @f0() #0 {
; RV64I-LABEL: f0:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: addi sp, sp, -64
; RV64I-NEXT: .cfi_def_cfa_offset 64
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 0(sp)
; RV64I-NEXT: lbu a0, 0(sp)
; RV64I-NEXT: addi sp, sp, 64
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f0:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: addi sp, sp, -64
; RV32I-NEXT: .cfi_def_cfa_offset 64
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 0(sp)
; RV32I-NEXT: lbu a0, 0(sp)
; RV32I-NEXT: addi sp, sp, 64
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 64
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
define i8 @f1() #0 {
; RV64I-LABEL: f1:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 1
; RV64I-NEXT: sub sp, sp, a0
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: .cfi_def_cfa_offset 4096
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: .cfi_def_cfa_offset 4112
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 16(sp)
; RV64I-NEXT: lbu a0, 16(sp)
; RV64I-NEXT: lui a1, 1
; RV64I-NEXT: addi a1, a1, 16
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f1:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 1
; RV32I-NEXT: sub sp, sp, a0
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: .cfi_def_cfa_offset 4096
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: .cfi_def_cfa_offset 4112
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 16(sp)
; RV32I-NEXT: lbu a0, 16(sp)
; RV32I-NEXT: lui a1, 1
; RV32I-NEXT: addi a1, a1, 16
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 4096
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
define i8 @f2() #0 {
; RV64I-LABEL: f2:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 16
; RV64I-NEXT: sub t1, sp, a0
; RV64I-NEXT: .cfi_def_cfa t1, 65536
; RV64I-NEXT: lui t2, 1
; RV64I-NEXT: .LBB2_1: # %entry
; RV64I-NEXT: # =>This Inner Loop Header: Depth=1
; RV64I-NEXT: sub sp, sp, t2
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: bne sp, t1, .LBB2_1
; RV64I-NEXT: # %bb.2: # %entry
; RV64I-NEXT: .cfi_def_cfa_register sp
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: .cfi_def_cfa_offset 65552
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 16(sp)
; RV64I-NEXT: lbu a0, 16(sp)
; RV64I-NEXT: lui a1, 16
; RV64I-NEXT: addi a1, a1, 16
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f2:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 16
; RV32I-NEXT: sub t1, sp, a0
; RV32I-NEXT: .cfi_def_cfa t1, 65536
; RV32I-NEXT: lui t2, 1
; RV32I-NEXT: .LBB2_1: # %entry
; RV32I-NEXT: # =>This Inner Loop Header: Depth=1
; RV32I-NEXT: sub sp, sp, t2
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: bne sp, t1, .LBB2_1
; RV32I-NEXT: # %bb.2: # %entry
; RV32I-NEXT: .cfi_def_cfa_register sp
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: .cfi_def_cfa_offset 65552
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 16(sp)
; RV32I-NEXT: lbu a0, 16(sp)
; RV32I-NEXT: lui a1, 16
; RV32I-NEXT: addi a1, a1, 16
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 65536
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
define i8 @f3() #0 "stack-probe-size"="32768" {
; RV64I-LABEL: f3:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 8
; RV64I-NEXT: sub sp, sp, a0
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: .cfi_def_cfa_offset 32768
; RV64I-NEXT: lui a0, 8
; RV64I-NEXT: sub sp, sp, a0
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: .cfi_def_cfa_offset 65536
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: .cfi_def_cfa_offset 65552
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 16(sp)
; RV64I-NEXT: lbu a0, 16(sp)
; RV64I-NEXT: lui a1, 16
; RV64I-NEXT: addi a1, a1, 16
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f3:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 8
; RV32I-NEXT: sub sp, sp, a0
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: .cfi_def_cfa_offset 32768
; RV32I-NEXT: lui a0, 8
; RV32I-NEXT: sub sp, sp, a0
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: .cfi_def_cfa_offset 65536
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: .cfi_def_cfa_offset 65552
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 16(sp)
; RV32I-NEXT: lbu a0, 16(sp)
; RV32I-NEXT: lui a1, 16
; RV32I-NEXT: addi a1, a1, 16
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 65536
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
; Same as f2, but without protection.
define i8 @f4() {
; RV64I-LABEL: f4:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 16
; RV64I-NEXT: addi a0, a0, 16
; RV64I-NEXT: sub sp, sp, a0
; RV64I-NEXT: .cfi_def_cfa_offset 65552
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 16(sp)
; RV64I-NEXT: lbu a0, 16(sp)
; RV64I-NEXT: lui a1, 16
; RV64I-NEXT: addi a1, a1, 16
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f4:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 16
; RV32I-NEXT: addi a0, a0, 16
; RV32I-NEXT: sub sp, sp, a0
; RV32I-NEXT: .cfi_def_cfa_offset 65552
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 16(sp)
; RV32I-NEXT: lbu a0, 16(sp)
; RV32I-NEXT: lui a1, 16
; RV32I-NEXT: addi a1, a1, 16
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 65536
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
define i8 @f5() #0 "stack-probe-size"="65536" {
; RV64I-LABEL: f5:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 256
; RV64I-NEXT: sub t1, sp, a0
; RV64I-NEXT: .cfi_def_cfa t1, 1048576
; RV64I-NEXT: lui t2, 16
; RV64I-NEXT: .LBB5_1: # %entry
; RV64I-NEXT: # =>This Inner Loop Header: Depth=1
; RV64I-NEXT: sub sp, sp, t2
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: bne sp, t1, .LBB5_1
; RV64I-NEXT: # %bb.2: # %entry
; RV64I-NEXT: .cfi_def_cfa_register sp
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: .cfi_def_cfa_offset 1048592
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 16(sp)
; RV64I-NEXT: lbu a0, 16(sp)
; RV64I-NEXT: lui a1, 256
; RV64I-NEXT: addi a1, a1, 16
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f5:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 256
; RV32I-NEXT: sub t1, sp, a0
; RV32I-NEXT: .cfi_def_cfa t1, 1048576
; RV32I-NEXT: lui t2, 16
; RV32I-NEXT: .LBB5_1: # %entry
; RV32I-NEXT: # =>This Inner Loop Header: Depth=1
; RV32I-NEXT: sub sp, sp, t2
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: bne sp, t1, .LBB5_1
; RV32I-NEXT: # %bb.2: # %entry
; RV32I-NEXT: .cfi_def_cfa_register sp
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: .cfi_def_cfa_offset 1048592
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 16(sp)
; RV32I-NEXT: lbu a0, 16(sp)
; RV32I-NEXT: lui a1, 256
; RV32I-NEXT: addi a1, a1, 16
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 1048576
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
define i8 @f6() #0 {
; RV64I-LABEL: f6:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 262144
; RV64I-NEXT: sub t1, sp, a0
; RV64I-NEXT: .cfi_def_cfa t1, 1073741824
; RV64I-NEXT: lui t2, 1
; RV64I-NEXT: .LBB6_1: # %entry
; RV64I-NEXT: # =>This Inner Loop Header: Depth=1
; RV64I-NEXT: sub sp, sp, t2
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: bne sp, t1, .LBB6_1
; RV64I-NEXT: # %bb.2: # %entry
; RV64I-NEXT: .cfi_def_cfa_register sp
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: .cfi_def_cfa_offset 1073741840
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 16(sp)
; RV64I-NEXT: lbu a0, 16(sp)
; RV64I-NEXT: lui a1, 262144
; RV64I-NEXT: addi a1, a1, 16
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f6:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 262144
; RV32I-NEXT: sub t1, sp, a0
; RV32I-NEXT: .cfi_def_cfa t1, 1073741824
; RV32I-NEXT: lui t2, 1
; RV32I-NEXT: .LBB6_1: # %entry
; RV32I-NEXT: # =>This Inner Loop Header: Depth=1
; RV32I-NEXT: sub sp, sp, t2
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: bne sp, t1, .LBB6_1
; RV32I-NEXT: # %bb.2: # %entry
; RV32I-NEXT: .cfi_def_cfa_register sp
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: .cfi_def_cfa_offset 1073741840
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 16(sp)
; RV32I-NEXT: lbu a0, 16(sp)
; RV32I-NEXT: lui a1, 262144
; RV32I-NEXT: addi a1, a1, 16
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 1073741824
%b = getelementptr inbounds i8, ptr %a, i64 63
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
define i8 @f7() #0 "stack-probe-size"="65536" {
; RV64I-LABEL: f7:
; RV64I: # %bb.0: # %entry
; RV64I-NEXT: lui a0, 244128
; RV64I-NEXT: sub t1, sp, a0
; RV64I-NEXT: .cfi_def_cfa t1, 999948288
; RV64I-NEXT: lui t2, 16
; RV64I-NEXT: .LBB7_1: # %entry
; RV64I-NEXT: # =>This Inner Loop Header: Depth=1
; RV64I-NEXT: sub sp, sp, t2
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: bne sp, t1, .LBB7_1
; RV64I-NEXT: # %bb.2: # %entry
; RV64I-NEXT: .cfi_def_cfa_register sp
; RV64I-NEXT: lui a0, 13
; RV64I-NEXT: addi a0, a0, -1520
; RV64I-NEXT: sub sp, sp, a0
; RV64I-NEXT: .cfi_def_cfa_offset 1000000016
; RV64I-NEXT: li a0, 3
; RV64I-NEXT: sb a0, 9(sp)
; RV64I-NEXT: lbu a0, 9(sp)
; RV64I-NEXT: lui a1, 244141
; RV64I-NEXT: addi a1, a1, -1520
; RV64I-NEXT: add sp, sp, a1
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f7:
; RV32I: # %bb.0: # %entry
; RV32I-NEXT: lui a0, 244128
; RV32I-NEXT: sub t1, sp, a0
; RV32I-NEXT: .cfi_def_cfa t1, 999948288
; RV32I-NEXT: lui t2, 16
; RV32I-NEXT: .LBB7_1: # %entry
; RV32I-NEXT: # =>This Inner Loop Header: Depth=1
; RV32I-NEXT: sub sp, sp, t2
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: bne sp, t1, .LBB7_1
; RV32I-NEXT: # %bb.2: # %entry
; RV32I-NEXT: .cfi_def_cfa_register sp
; RV32I-NEXT: lui a0, 13
; RV32I-NEXT: addi a0, a0, -1520
; RV32I-NEXT: sub sp, sp, a0
; RV32I-NEXT: .cfi_def_cfa_offset 1000000016
; RV32I-NEXT: li a0, 3
; RV32I-NEXT: sb a0, 9(sp)
; RV32I-NEXT: lbu a0, 9(sp)
; RV32I-NEXT: lui a1, 244141
; RV32I-NEXT: addi a1, a1, -1520
; RV32I-NEXT: add sp, sp, a1
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
entry:
%a = alloca i8, i64 1000000007
%b = getelementptr inbounds i8, ptr %a, i64 101
store volatile i8 3, ptr %a
%c = load volatile i8, ptr %a
ret i8 %c
}
; alloca + align < probe_size
define i32 @f8(i64 %i) local_unnamed_addr #0 {
; RV64I-LABEL: f8:
; RV64I: # %bb.0:
; RV64I-NEXT: addi sp, sp, -832
; RV64I-NEXT: .cfi_def_cfa_offset 832
; RV64I-NEXT: sd ra, 824(sp) # 8-byte Folded Spill
; RV64I-NEXT: sd s0, 816(sp) # 8-byte Folded Spill
; RV64I-NEXT: .cfi_offset ra, -8
; RV64I-NEXT: .cfi_offset s0, -16
; RV64I-NEXT: addi s0, sp, 832
; RV64I-NEXT: .cfi_def_cfa s0, 0
; RV64I-NEXT: andi sp, sp, -64
; RV64I-NEXT: slli a0, a0, 2
; RV64I-NEXT: mv a1, sp
; RV64I-NEXT: add a0, a1, a0
; RV64I-NEXT: li a1, 1
; RV64I-NEXT: sw a1, 0(a0)
; RV64I-NEXT: lw a0, 0(sp)
; RV64I-NEXT: addi sp, s0, -832
; RV64I-NEXT: .cfi_def_cfa sp, 832
; RV64I-NEXT: ld ra, 824(sp) # 8-byte Folded Reload
; RV64I-NEXT: ld s0, 816(sp) # 8-byte Folded Reload
; RV64I-NEXT: .cfi_restore ra
; RV64I-NEXT: .cfi_restore s0
; RV64I-NEXT: addi sp, sp, 832
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f8:
; RV32I: # %bb.0:
; RV32I-NEXT: addi sp, sp, -832
; RV32I-NEXT: .cfi_def_cfa_offset 832
; RV32I-NEXT: sw ra, 828(sp) # 4-byte Folded Spill
; RV32I-NEXT: sw s0, 824(sp) # 4-byte Folded Spill
; RV32I-NEXT: .cfi_offset ra, -4
; RV32I-NEXT: .cfi_offset s0, -8
; RV32I-NEXT: addi s0, sp, 832
; RV32I-NEXT: .cfi_def_cfa s0, 0
; RV32I-NEXT: andi sp, sp, -64
; RV32I-NEXT: slli a0, a0, 2
; RV32I-NEXT: mv a1, sp
; RV32I-NEXT: add a0, a1, a0
; RV32I-NEXT: li a1, 1
; RV32I-NEXT: sw a1, 0(a0)
; RV32I-NEXT: lw a0, 0(sp)
; RV32I-NEXT: addi sp, s0, -832
; RV32I-NEXT: .cfi_def_cfa sp, 832
; RV32I-NEXT: lw ra, 828(sp) # 4-byte Folded Reload
; RV32I-NEXT: lw s0, 824(sp) # 4-byte Folded Reload
; RV32I-NEXT: .cfi_restore ra
; RV32I-NEXT: .cfi_restore s0
; RV32I-NEXT: addi sp, sp, 832
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
%a = alloca i32, i32 200, align 64
%b = getelementptr inbounds i32, ptr %a, i64 %i
store volatile i32 1, ptr %b
%c = load volatile i32, ptr %a
ret i32 %c
}
; alloca > probe_size, align > probe_size
define i32 @f9(i64 %i) local_unnamed_addr #0 {
; RV64I-LABEL: f9:
; RV64I: # %bb.0:
; RV64I-NEXT: addi sp, sp, -2032
; RV64I-NEXT: .cfi_def_cfa_offset 2032
; RV64I-NEXT: sd ra, 2024(sp) # 8-byte Folded Spill
; RV64I-NEXT: sd s0, 2016(sp) # 8-byte Folded Spill
; RV64I-NEXT: .cfi_offset ra, -8
; RV64I-NEXT: .cfi_offset s0, -16
; RV64I-NEXT: addi s0, sp, 2032
; RV64I-NEXT: .cfi_def_cfa s0, 0
; RV64I-NEXT: lui a1, 1
; RV64I-NEXT: sub sp, sp, a1
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: sub sp, sp, a1
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: andi sp, sp, -2048
; RV64I-NEXT: slli a0, a0, 2
; RV64I-NEXT: addi a1, sp, 2047
; RV64I-NEXT: addi a1, a1, 1
; RV64I-NEXT: add a0, a1, a0
; RV64I-NEXT: li a1, 1
; RV64I-NEXT: sw a1, 0(a0)
; RV64I-NEXT: lui a0, 1
; RV64I-NEXT: add a0, sp, a0
; RV64I-NEXT: lw a0, -2048(a0)
; RV64I-NEXT: addi sp, s0, -2032
; RV64I-NEXT: .cfi_def_cfa sp, 2032
; RV64I-NEXT: ld ra, 2024(sp) # 8-byte Folded Reload
; RV64I-NEXT: ld s0, 2016(sp) # 8-byte Folded Reload
; RV64I-NEXT: .cfi_restore ra
; RV64I-NEXT: .cfi_restore s0
; RV64I-NEXT: addi sp, sp, 2032
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f9:
; RV32I: # %bb.0:
; RV32I-NEXT: addi sp, sp, -2032
; RV32I-NEXT: .cfi_def_cfa_offset 2032
; RV32I-NEXT: sw ra, 2028(sp) # 4-byte Folded Spill
; RV32I-NEXT: sw s0, 2024(sp) # 4-byte Folded Spill
; RV32I-NEXT: .cfi_offset ra, -4
; RV32I-NEXT: .cfi_offset s0, -8
; RV32I-NEXT: addi s0, sp, 2032
; RV32I-NEXT: .cfi_def_cfa s0, 0
; RV32I-NEXT: lui a1, 1
; RV32I-NEXT: sub sp, sp, a1
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: sub sp, sp, a1
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: andi sp, sp, -2048
; RV32I-NEXT: slli a0, a0, 2
; RV32I-NEXT: addi a1, sp, 2047
; RV32I-NEXT: addi a1, a1, 1
; RV32I-NEXT: add a0, a1, a0
; RV32I-NEXT: li a1, 1
; RV32I-NEXT: sw a1, 0(a0)
; RV32I-NEXT: lui a0, 1
; RV32I-NEXT: add a0, sp, a0
; RV32I-NEXT: lw a0, -2048(a0)
; RV32I-NEXT: addi sp, s0, -2032
; RV32I-NEXT: .cfi_def_cfa sp, 2032
; RV32I-NEXT: lw ra, 2028(sp) # 4-byte Folded Reload
; RV32I-NEXT: lw s0, 2024(sp) # 4-byte Folded Reload
; RV32I-NEXT: .cfi_restore ra
; RV32I-NEXT: .cfi_restore s0
; RV32I-NEXT: addi sp, sp, 2032
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
%a = alloca i32, i32 2000, align 2048
%b = getelementptr inbounds i32, ptr %a, i64 %i
store volatile i32 1, ptr %b
%c = load volatile i32, ptr %a
ret i32 %c
}
; alloca < probe_size, align < probe_size, alloca + align > probe_size
define i32 @f10(i64 %i) local_unnamed_addr #0 {
; RV64I-LABEL: f10:
; RV64I: # %bb.0:
; RV64I-NEXT: addi sp, sp, -2032
; RV64I-NEXT: .cfi_def_cfa_offset 2032
; RV64I-NEXT: sd ra, 2024(sp) # 8-byte Folded Spill
; RV64I-NEXT: sd s0, 2016(sp) # 8-byte Folded Spill
; RV64I-NEXT: .cfi_offset ra, -8
; RV64I-NEXT: .cfi_offset s0, -16
; RV64I-NEXT: addi s0, sp, 2032
; RV64I-NEXT: .cfi_def_cfa s0, 0
; RV64I-NEXT: addi sp, sp, -2048
; RV64I-NEXT: addi sp, sp, -1040
; RV64I-NEXT: andi sp, sp, -1024
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: slli a0, a0, 2
; RV64I-NEXT: addi a1, sp, 1024
; RV64I-NEXT: add a0, a1, a0
; RV64I-NEXT: li a1, 1
; RV64I-NEXT: sw a1, 0(a0)
; RV64I-NEXT: lw a0, 1024(sp)
; RV64I-NEXT: addi sp, s0, -2032
; RV64I-NEXT: .cfi_def_cfa sp, 2032
; RV64I-NEXT: ld ra, 2024(sp) # 8-byte Folded Reload
; RV64I-NEXT: ld s0, 2016(sp) # 8-byte Folded Reload
; RV64I-NEXT: .cfi_restore ra
; RV64I-NEXT: .cfi_restore s0
; RV64I-NEXT: addi sp, sp, 2032
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f10:
; RV32I: # %bb.0:
; RV32I-NEXT: addi sp, sp, -2032
; RV32I-NEXT: .cfi_def_cfa_offset 2032
; RV32I-NEXT: sw ra, 2028(sp) # 4-byte Folded Spill
; RV32I-NEXT: sw s0, 2024(sp) # 4-byte Folded Spill
; RV32I-NEXT: .cfi_offset ra, -4
; RV32I-NEXT: .cfi_offset s0, -8
; RV32I-NEXT: addi s0, sp, 2032
; RV32I-NEXT: .cfi_def_cfa s0, 0
; RV32I-NEXT: addi sp, sp, -2048
; RV32I-NEXT: addi sp, sp, -1040
; RV32I-NEXT: andi sp, sp, -1024
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: slli a0, a0, 2
; RV32I-NEXT: addi a1, sp, 1024
; RV32I-NEXT: add a0, a1, a0
; RV32I-NEXT: li a1, 1
; RV32I-NEXT: sw a1, 0(a0)
; RV32I-NEXT: lw a0, 1024(sp)
; RV32I-NEXT: addi sp, s0, -2032
; RV32I-NEXT: .cfi_def_cfa sp, 2032
; RV32I-NEXT: lw ra, 2028(sp) # 4-byte Folded Reload
; RV32I-NEXT: lw s0, 2024(sp) # 4-byte Folded Reload
; RV32I-NEXT: .cfi_restore ra
; RV32I-NEXT: .cfi_restore s0
; RV32I-NEXT: addi sp, sp, 2032
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
%a = alloca i32, i32 1000, align 1024
%b = getelementptr inbounds i32, ptr %a, i64 %i
store volatile i32 1, ptr %b
%c = load volatile i32, ptr %a
ret i32 %c
}
define void @f11(i32 %vla_size, i64 %i) #0 {
; RV64I-LABEL: f11:
; RV64I: # %bb.0:
; RV64I-NEXT: addi sp, sp, -2032
; RV64I-NEXT: .cfi_def_cfa_offset 2032
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: sd ra, 2024(sp) # 8-byte Folded Spill
; RV64I-NEXT: sd s0, 2016(sp) # 8-byte Folded Spill
; RV64I-NEXT: sd s1, 2008(sp) # 8-byte Folded Spill
; RV64I-NEXT: .cfi_offset ra, -8
; RV64I-NEXT: .cfi_offset s0, -16
; RV64I-NEXT: .cfi_offset s1, -24
; RV64I-NEXT: addi s0, sp, 2032
; RV64I-NEXT: .cfi_def_cfa s0, 0
; RV64I-NEXT: lui a2, 15
; RV64I-NEXT: sub t1, sp, a2
; RV64I-NEXT: lui t2, 1
; RV64I-NEXT: .LBB11_1: # =>This Inner Loop Header: Depth=1
; RV64I-NEXT: sub sp, sp, t2
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: bne sp, t1, .LBB11_1
; RV64I-NEXT: # %bb.2:
; RV64I-NEXT: addi sp, sp, -2048
; RV64I-NEXT: addi sp, sp, -16
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: srli a2, sp, 15
; RV64I-NEXT: slli sp, a2, 15
; RV64I-NEXT: mv s1, sp
; RV64I-NEXT: slli a1, a1, 2
; RV64I-NEXT: lui a2, 8
; RV64I-NEXT: add a2, s1, a2
; RV64I-NEXT: add a1, a2, a1
; RV64I-NEXT: li a2, 1
; RV64I-NEXT: slli a0, a0, 32
; RV64I-NEXT: srli a0, a0, 32
; RV64I-NEXT: sw a2, 0(a1)
; RV64I-NEXT: addi a0, a0, 15
; RV64I-NEXT: andi a0, a0, -16
; RV64I-NEXT: sub a0, sp, a0
; RV64I-NEXT: andi a0, a0, -2048
; RV64I-NEXT: lui a1, 1
; RV64I-NEXT: .LBB11_3: # =>This Inner Loop Header: Depth=1
; RV64I-NEXT: sub sp, sp, a1
; RV64I-NEXT: sd zero, 0(sp)
; RV64I-NEXT: blt a0, sp, .LBB11_3
; RV64I-NEXT: # %bb.4:
; RV64I-NEXT: mv sp, a0
; RV64I-NEXT: lbu zero, 0(a0)
; RV64I-NEXT: addi sp, s0, -2032
; RV64I-NEXT: .cfi_def_cfa sp, 2032
; RV64I-NEXT: ld ra, 2024(sp) # 8-byte Folded Reload
; RV64I-NEXT: ld s0, 2016(sp) # 8-byte Folded Reload
; RV64I-NEXT: ld s1, 2008(sp) # 8-byte Folded Reload
; RV64I-NEXT: .cfi_restore ra
; RV64I-NEXT: .cfi_restore s0
; RV64I-NEXT: .cfi_restore s1
; RV64I-NEXT: addi sp, sp, 2032
; RV64I-NEXT: .cfi_def_cfa_offset 0
; RV64I-NEXT: ret
;
; RV32I-LABEL: f11:
; RV32I: # %bb.0:
; RV32I-NEXT: addi sp, sp, -2032
; RV32I-NEXT: .cfi_def_cfa_offset 2032
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: sw ra, 2028(sp) # 4-byte Folded Spill
; RV32I-NEXT: sw s0, 2024(sp) # 4-byte Folded Spill
; RV32I-NEXT: sw s1, 2020(sp) # 4-byte Folded Spill
; RV32I-NEXT: .cfi_offset ra, -4
; RV32I-NEXT: .cfi_offset s0, -8
; RV32I-NEXT: .cfi_offset s1, -12
; RV32I-NEXT: addi s0, sp, 2032
; RV32I-NEXT: .cfi_def_cfa s0, 0
; RV32I-NEXT: lui a2, 15
; RV32I-NEXT: sub t1, sp, a2
; RV32I-NEXT: lui t2, 1
; RV32I-NEXT: .LBB11_1: # =>This Inner Loop Header: Depth=1
; RV32I-NEXT: sub sp, sp, t2
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: bne sp, t1, .LBB11_1
; RV32I-NEXT: # %bb.2:
; RV32I-NEXT: addi sp, sp, -2048
; RV32I-NEXT: addi sp, sp, -16
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: srli a2, sp, 15
; RV32I-NEXT: slli sp, a2, 15
; RV32I-NEXT: mv s1, sp
; RV32I-NEXT: slli a1, a1, 2
; RV32I-NEXT: lui a2, 8
; RV32I-NEXT: add a2, s1, a2
; RV32I-NEXT: add a1, a2, a1
; RV32I-NEXT: li a2, 1
; RV32I-NEXT: addi a0, a0, 15
; RV32I-NEXT: andi a0, a0, -16
; RV32I-NEXT: sw a2, 0(a1)
; RV32I-NEXT: sub a0, sp, a0
; RV32I-NEXT: andi a0, a0, -2048
; RV32I-NEXT: lui a1, 1
; RV32I-NEXT: .LBB11_3: # =>This Inner Loop Header: Depth=1
; RV32I-NEXT: sub sp, sp, a1
; RV32I-NEXT: sw zero, 0(sp)
; RV32I-NEXT: blt a0, sp, .LBB11_3
; RV32I-NEXT: # %bb.4:
; RV32I-NEXT: mv sp, a0
; RV32I-NEXT: lbu zero, 0(a0)
; RV32I-NEXT: addi sp, s0, -2032
; RV32I-NEXT: .cfi_def_cfa sp, 2032
; RV32I-NEXT: lw ra, 2028(sp) # 4-byte Folded Reload
; RV32I-NEXT: lw s0, 2024(sp) # 4-byte Folded Reload
; RV32I-NEXT: lw s1, 2020(sp) # 4-byte Folded Reload
; RV32I-NEXT: .cfi_restore ra
; RV32I-NEXT: .cfi_restore s0
; RV32I-NEXT: .cfi_restore s1
; RV32I-NEXT: addi sp, sp, 2032
; RV32I-NEXT: .cfi_def_cfa_offset 0
; RV32I-NEXT: ret
%a = alloca i32, i32 4096, align 32768
%b = getelementptr inbounds i32, ptr %a, i64 %i
store volatile i32 1, ptr %b
%1 = zext i32 %vla_size to i64
%vla = alloca i8, i64 %1, align 2048
%2 = load volatile i8, ptr %vla, align 2048
ret void
}
attributes #0 = { "probe-stack"="inline-asm" }