diff --git a/llvm/test/CodeGen/RISCV/fold-addi-loadstore.ll b/llvm/test/CodeGen/RISCV/fold-addi-loadstore.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/RISCV/fold-addi-loadstore.ll @@ -0,0 +1,261 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py +; RUN: llc -mtriple=riscv32 -verify-machineinstrs < %s \ +; RUN: | FileCheck -check-prefix=RV32I %s +; RUN: llc -mtriple=riscv64 -verify-machineinstrs < %s \ +; RUN: | FileCheck -check-prefix=RV64I %s + +; We can often fold an ADDI into the offset of load/store instructions: +; (load (addi base, off1), off2) -> (load base, off1+off2) +; (store val, (addi base, off1), off2) -> (store val, base, off1+off2) +; This is possible when the off1+off2 continues to fit the 12-bit immediate. +; Check if we do the fold under various conditions. If off1 is (the low part of) +; an address the fold's safety depends on the variable's alignment. + +@g_0 = global i64 0 +@g_1 = global i64 0, align 1 +@g_2 = global i64 0, align 2 +@g_4 = global i64 0, align 4 +@g_8 = global i64 0, align 8 +@g_16 = global i64 0, align 16 + +define i64 @load_g_0() nounwind { +; RV32I-LABEL: load_g_0: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a1, %hi(g_0) +; RV32I-NEXT: lw a0, %lo(g_0)(a1) +; RV32I-NEXT: addi a1, a1, %lo(g_0) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_g_0: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_0) +; RV64I-NEXT: ld a0, %lo(g_0)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @g_0 + ret i64 %0 +} + +define i64 @load_g_1() nounwind { +; RV32I-LABEL: load_g_1: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a1, %hi(g_1) +; RV32I-NEXT: lw a0, %lo(g_1)(a1) +; RV32I-NEXT: addi a1, a1, %lo(g_1) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_g_1: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_1) +; RV64I-NEXT: ld a0, %lo(g_1)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @g_1 + ret i64 %0 +} + +define i64 @load_g_2() nounwind { +; RV32I-LABEL: load_g_2: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a1, %hi(g_2) +; RV32I-NEXT: lw a0, %lo(g_2)(a1) +; RV32I-NEXT: addi a1, a1, %lo(g_2) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_g_2: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_2) +; RV64I-NEXT: ld a0, %lo(g_2)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @g_2 + ret i64 %0 +} + +define i64 @load_g_4() nounwind { +; RV32I-LABEL: load_g_4: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a1, %hi(g_4) +; RV32I-NEXT: lw a0, %lo(g_4)(a1) +; RV32I-NEXT: addi a1, a1, %lo(g_4) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_g_4: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_4) +; RV64I-NEXT: ld a0, %lo(g_4)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @g_4 + ret i64 %0 +} + +define i64 @load_g_8() nounwind { +; RV32I-LABEL: load_g_8: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a1, %hi(g_8) +; RV32I-NEXT: lw a0, %lo(g_8)(a1) +; RV32I-NEXT: addi a1, a1, %lo(g_8) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_g_8: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_8) +; RV64I-NEXT: ld a0, %lo(g_8)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @g_8 + ret i64 %0 +} + +define i64 @load_g_16() nounwind { +; RV32I-LABEL: load_g_16: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a1, %hi(g_16) +; RV32I-NEXT: lw a0, %lo(g_16)(a1) +; RV32I-NEXT: addi a1, a1, %lo(g_16) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_g_16: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_16) +; RV64I-NEXT: ld a0, %lo(g_16)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @g_16 + ret i64 %0 +} + +define void @store_g_4() nounwind { +; RV32I-LABEL: store_g_4: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a0, %hi(g_4) +; RV32I-NEXT: sw zero, %lo(g_4)(a0) +; RV32I-NEXT: addi a0, a0, %lo(g_4) +; RV32I-NEXT: sw zero, 4(a0) +; RV32I-NEXT: ret +; +; RV64I-LABEL: store_g_4: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_4) +; RV64I-NEXT: sd zero, %lo(g_4)(a0) +; RV64I-NEXT: ret +entry: + store i64 0, i64* @g_4 + ret void +} + +define void @store_g_8() nounwind { +; RV32I-LABEL: store_g_8: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a0, %hi(g_8) +; RV32I-NEXT: sw zero, %lo(g_8)(a0) +; RV32I-NEXT: addi a0, a0, %lo(g_8) +; RV32I-NEXT: sw zero, 4(a0) +; RV32I-NEXT: ret +; +; RV64I-LABEL: store_g_8: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(g_8) +; RV64I-NEXT: sd zero, %lo(g_8)(a0) +; RV64I-NEXT: ret +entry: + store i64 0, i64* @g_8 + ret void +} + +; Check for folds in accesses to the second element of an i64 array. + +@ga_8 = dso_local local_unnamed_addr global [2 x i64] zeroinitializer, align 8 +@ga_16 = dso_local local_unnamed_addr global [2 x i64] zeroinitializer, align 16 + +define i64 @load_ga_8() nounwind { +; RV32I-LABEL: load_ga_8: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a0, %hi(ga_8) +; RV32I-NEXT: addi a1, a0, %lo(ga_8) +; RV32I-NEXT: lw a0, 8(a1) +; RV32I-NEXT: lw a1, 12(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_ga_8: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(ga_8+8) +; RV64I-NEXT: ld a0, %lo(ga_8+8)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* getelementptr inbounds ([2 x i64], [2 x i64]* @ga_8, i32 0, i32 1) + ret i64 %0 +} + +define i64 @load_ga_16() nounwind { +; RV32I-LABEL: load_ga_16: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a0, %hi(ga_16) +; RV32I-NEXT: addi a1, a0, %lo(ga_16) +; RV32I-NEXT: lw a0, 8(a1) +; RV32I-NEXT: lw a1, 12(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_ga_16: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %hi(ga_16+8) +; RV64I-NEXT: ld a0, %lo(ga_16+8)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* getelementptr inbounds ([2 x i64], [2 x i64]* @ga_16, i32 0, i32 1) + ret i64 %0 +} + +; Check for folds in accesses to thread-local variables. + +@tl_4 = thread_local global i64 0, align 4 +@tl_8 = thread_local global i64 0, align 8 + +define i64 @load_tl_4() nounwind { +; RV32I-LABEL: load_tl_4: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a0, %tprel_hi(tl_4) +; RV32I-NEXT: add a1, a0, tp, %tprel_add(tl_4) +; RV32I-NEXT: lw a0, %tprel_lo(tl_4)(a1) +; RV32I-NEXT: addi a1, a1, %tprel_lo(tl_4) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_tl_4: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %tprel_hi(tl_4) +; RV64I-NEXT: add a0, a0, tp, %tprel_add(tl_4) +; RV64I-NEXT: ld a0, %tprel_lo(tl_4)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @tl_4 + ret i64 %0 +} + +define i64 @load_tl_8() nounwind { +; RV32I-LABEL: load_tl_8: +; RV32I: # %bb.0: # %entry +; RV32I-NEXT: lui a0, %tprel_hi(tl_8) +; RV32I-NEXT: add a1, a0, tp, %tprel_add(tl_8) +; RV32I-NEXT: lw a0, %tprel_lo(tl_8)(a1) +; RV32I-NEXT: addi a1, a1, %tprel_lo(tl_8) +; RV32I-NEXT: lw a1, 4(a1) +; RV32I-NEXT: ret +; +; RV64I-LABEL: load_tl_8: +; RV64I: # %bb.0: # %entry +; RV64I-NEXT: lui a0, %tprel_hi(tl_8) +; RV64I-NEXT: add a0, a0, tp, %tprel_add(tl_8) +; RV64I-NEXT: ld a0, %tprel_lo(tl_8)(a0) +; RV64I-NEXT: ret +entry: + %0 = load i64, i64* @tl_8 + ret i64 %0 +}