Index: llvm/lib/Target/RISCV/RISCV.td =================================================================== --- llvm/lib/Target/RISCV/RISCV.td +++ llvm/lib/Target/RISCV/RISCV.td @@ -430,6 +430,11 @@ def FeatureSaveRestore : SubtargetFeature<"save-restore", "EnableSaveRestore", "true", "Enable save/restore.">; +def FeatureUnalignedMem + : SubtargetFeature<"unaligned-mem", "EnableUnalignedMem", + "true", "Implements unaligned scalar loads and stores">; + + def TuneNoDefaultUnroll : SubtargetFeature<"no-default-unroll", "EnableDefaultUnroll", "false", "Disable default unroll preference.">; Index: llvm/lib/Target/RISCV/RISCVISelLowering.cpp =================================================================== --- llvm/lib/Target/RISCV/RISCVISelLowering.cpp +++ llvm/lib/Target/RISCV/RISCVISelLowering.cpp @@ -11807,8 +11807,9 @@ EVT VT, unsigned AddrSpace, Align Alignment, MachineMemOperand::Flags Flags, bool *Fast) const { if (!VT.isVector()) - return false; + return Subtarget.enableUnalignedMem(); + // All vector implementations must support element alignment EVT ElemVT = VT.getVectorElementType(); if (Alignment >= ElemVT.getStoreSize()) { if (Fast) @@ -11816,7 +11817,7 @@ return true; } - return false; + return Subtarget.enableUnalignedMem(); } bool RISCVTargetLowering::splitValueIntoRegisterParts( Index: llvm/lib/Target/RISCV/RISCVSubtarget.h =================================================================== --- llvm/lib/Target/RISCV/RISCVSubtarget.h +++ llvm/lib/Target/RISCV/RISCVSubtarget.h @@ -92,6 +92,7 @@ bool EnableRVCHintInstrs = true; bool EnableDefaultUnroll = true; bool EnableSaveRestore = false; + bool EnableUnalignedMem = false; unsigned XLen = 32; unsigned ZvlLen = 0; MVT XLenVT = MVT::i32; @@ -182,6 +183,7 @@ bool enableRVCHintInstrs() const { return EnableRVCHintInstrs; } bool enableDefaultUnroll() const { return EnableDefaultUnroll; } bool enableSaveRestore() const { return EnableSaveRestore; } + bool enableUnalignedMem() const { return EnableUnalignedMem; } MVT getXLenVT() const { return XLenVT; } unsigned getXLen() const { return XLen; } unsigned getFLen() const { Index: llvm/test/CodeGen/RISCV/unaligned-load-store.ll =================================================================== --- llvm/test/CodeGen/RISCV/unaligned-load-store.ll +++ llvm/test/CodeGen/RISCV/unaligned-load-store.ll @@ -1,43 +1,60 @@ ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mtriple=riscv32 -verify-machineinstrs < %s \ -; RUN: | FileCheck -check-prefixes=BOTH,RV32I %s +; RUN: | FileCheck -check-prefixes=ALL,NOALIGN,RV32I %s ; RUN: llc -mtriple=riscv64 -verify-machineinstrs < %s \ -; RUN: | FileCheck -check-prefixes=BOTH,RV64I %s +; RUN: | FileCheck -check-prefixes=ALL,NOALIGN,RV64I %s +; RUN: llc -mtriple=riscv32 -mattr=+unaligned-mem -verify-machineinstrs < %s \ +; RUN: | FileCheck -check-prefixes=ALL,ALIGN,ALIGN-RV32I %s +; RUN: llc -mtriple=riscv64 -mattr=+unaligned-mem -verify-machineinstrs < %s \ +; RUN: | FileCheck -check-prefixes=ALL,ALIGN,ALIGN-RV64I %s ; A collection of cases showing codegen for unaligned loads and stores define i8 @load_i8(i8* %p) { -; BOTH-LABEL: load_i8: -; BOTH: # %bb.0: -; BOTH-NEXT: lb a0, 0(a0) -; BOTH-NEXT: ret +; ALL-LABEL: load_i8: +; ALL: # %bb.0: +; ALL-NEXT: lb a0, 0(a0) +; ALL-NEXT: ret %res = load i8, i8* %p, align 1 ret i8 %res } define i16 @load_i16(i16* %p) { -; BOTH-LABEL: load_i16: -; BOTH: # %bb.0: -; BOTH-NEXT: lb a1, 1(a0) -; BOTH-NEXT: lbu a0, 0(a0) -; BOTH-NEXT: slli a1, a1, 8 -; BOTH-NEXT: or a0, a1, a0 -; BOTH-NEXT: ret +; NOALIGN-LABEL: load_i16: +; NOALIGN: # %bb.0: +; NOALIGN-NEXT: lb a1, 1(a0) +; NOALIGN-NEXT: lbu a0, 0(a0) +; NOALIGN-NEXT: slli a1, a1, 8 +; NOALIGN-NEXT: or a0, a1, a0 +; NOALIGN-NEXT: ret +; +; ALIGN-LABEL: load_i16: +; ALIGN: # %bb.0: +; ALIGN-NEXT: lh a0, 0(a0) +; ALIGN-NEXT: ret %res = load i16, i16* %p, align 1 ret i16 %res } define i24 @load_i24(i24* %p) { -; BOTH-LABEL: load_i24: -; BOTH: # %bb.0: -; BOTH-NEXT: lbu a1, 1(a0) -; BOTH-NEXT: lbu a2, 0(a0) -; BOTH-NEXT: lb a0, 2(a0) -; BOTH-NEXT: slli a1, a1, 8 -; BOTH-NEXT: or a1, a1, a2 -; BOTH-NEXT: slli a0, a0, 16 -; BOTH-NEXT: or a0, a1, a0 -; BOTH-NEXT: ret +; NOALIGN-LABEL: load_i24: +; NOALIGN: # %bb.0: +; NOALIGN-NEXT: lbu a1, 1(a0) +; NOALIGN-NEXT: lbu a2, 0(a0) +; NOALIGN-NEXT: lb a0, 2(a0) +; NOALIGN-NEXT: slli a1, a1, 8 +; NOALIGN-NEXT: or a1, a1, a2 +; NOALIGN-NEXT: slli a0, a0, 16 +; NOALIGN-NEXT: or a0, a1, a0 +; NOALIGN-NEXT: ret +; +; ALIGN-LABEL: load_i24: +; ALIGN: # %bb.0: +; ALIGN-NEXT: lb a1, 2(a0) +; ALIGN-NEXT: lhu a0, 0(a0) +; ALIGN-NEXT: slli a1, a1, 16 +; ALIGN-NEXT: or a0, a0, a1 +; ALIGN-NEXT: ret %res = load i24, i24* %p, align 1 ret i24 %res } @@ -70,6 +87,11 @@ ; RV64I-NEXT: slli a0, a0, 16 ; RV64I-NEXT: or a0, a0, a1 ; RV64I-NEXT: ret +; +; ALIGN-LABEL: load_i32: +; ALIGN: # %bb.0: +; ALIGN-NEXT: lw a0, 0(a0) +; ALIGN-NEXT: ret %res = load i32, i32* %p, align 1 ret i32 %res } @@ -125,54 +147,83 @@ ; RV64I-NEXT: slli a0, a0, 32 ; RV64I-NEXT: or a0, a0, a1 ; RV64I-NEXT: ret +; +; ALIGN-RV32I-LABEL: load_i64: +; ALIGN-RV32I: # %bb.0: +; ALIGN-RV32I-NEXT: lw a2, 0(a0) +; ALIGN-RV32I-NEXT: lw a1, 4(a0) +; ALIGN-RV32I-NEXT: mv a0, a2 +; ALIGN-RV32I-NEXT: ret +; +; ALIGN-RV64I-LABEL: load_i64: +; ALIGN-RV64I: # %bb.0: +; ALIGN-RV64I-NEXT: ld a0, 0(a0) +; ALIGN-RV64I-NEXT: ret %res = load i64, i64* %p, align 1 ret i64 %res } define void @store_i8(i8* %p, i8 %v) { -; BOTH-LABEL: store_i8: -; BOTH: # %bb.0: -; BOTH-NEXT: sb a1, 0(a0) -; BOTH-NEXT: ret +; ALL-LABEL: store_i8: +; ALL: # %bb.0: +; ALL-NEXT: sb a1, 0(a0) +; ALL-NEXT: ret store i8 %v, i8* %p, align 1 ret void } define void @store_i16(i16* %p, i16 %v) { -; BOTH-LABEL: store_i16: -; BOTH: # %bb.0: -; BOTH-NEXT: sb a1, 0(a0) -; BOTH-NEXT: srli a1, a1, 8 -; BOTH-NEXT: sb a1, 1(a0) -; BOTH-NEXT: ret +; NOALIGN-LABEL: store_i16: +; NOALIGN: # %bb.0: +; NOALIGN-NEXT: sb a1, 0(a0) +; NOALIGN-NEXT: srli a1, a1, 8 +; NOALIGN-NEXT: sb a1, 1(a0) +; NOALIGN-NEXT: ret +; +; ALIGN-LABEL: store_i16: +; ALIGN: # %bb.0: +; ALIGN-NEXT: sh a1, 0(a0) +; ALIGN-NEXT: ret store i16 %v, i16* %p, align 1 ret void } define void @store_i24(i24* %p, i24 %v) { -; BOTH-LABEL: store_i24: -; BOTH: # %bb.0: -; BOTH-NEXT: sb a1, 0(a0) -; BOTH-NEXT: srli a2, a1, 8 -; BOTH-NEXT: sb a2, 1(a0) -; BOTH-NEXT: srli a1, a1, 16 -; BOTH-NEXT: sb a1, 2(a0) -; BOTH-NEXT: ret +; NOALIGN-LABEL: store_i24: +; NOALIGN: # %bb.0: +; NOALIGN-NEXT: sb a1, 0(a0) +; NOALIGN-NEXT: srli a2, a1, 8 +; NOALIGN-NEXT: sb a2, 1(a0) +; NOALIGN-NEXT: srli a1, a1, 16 +; NOALIGN-NEXT: sb a1, 2(a0) +; NOALIGN-NEXT: ret +; +; ALIGN-LABEL: store_i24: +; ALIGN: # %bb.0: +; ALIGN-NEXT: sh a1, 0(a0) +; ALIGN-NEXT: srli a1, a1, 16 +; ALIGN-NEXT: sb a1, 2(a0) +; ALIGN-NEXT: ret store i24 %v, i24* %p, align 1 ret void } define void @store_i32(i32* %p, i32 %v) { -; BOTH-LABEL: store_i32: -; BOTH: # %bb.0: -; BOTH-NEXT: sb a1, 0(a0) -; BOTH-NEXT: srli a2, a1, 24 -; BOTH-NEXT: sb a2, 3(a0) -; BOTH-NEXT: srli a2, a1, 16 -; BOTH-NEXT: sb a2, 2(a0) -; BOTH-NEXT: srli a1, a1, 8 -; BOTH-NEXT: sb a1, 1(a0) -; BOTH-NEXT: ret +; NOALIGN-LABEL: store_i32: +; NOALIGN: # %bb.0: +; NOALIGN-NEXT: sb a1, 0(a0) +; NOALIGN-NEXT: srli a2, a1, 24 +; NOALIGN-NEXT: sb a2, 3(a0) +; NOALIGN-NEXT: srli a2, a1, 16 +; NOALIGN-NEXT: sb a2, 2(a0) +; NOALIGN-NEXT: srli a1, a1, 8 +; NOALIGN-NEXT: sb a1, 1(a0) +; NOALIGN-NEXT: ret +; +; ALIGN-LABEL: store_i32: +; ALIGN: # %bb.0: +; ALIGN-NEXT: sw a1, 0(a0) +; ALIGN-NEXT: ret store i32 %v, i32* %p, align 1 ret void } @@ -214,6 +265,17 @@ ; RV64I-NEXT: srli a1, a1, 8 ; RV64I-NEXT: sb a1, 1(a0) ; RV64I-NEXT: ret +; +; ALIGN-RV32I-LABEL: store_i64: +; ALIGN-RV32I: # %bb.0: +; ALIGN-RV32I-NEXT: sw a2, 4(a0) +; ALIGN-RV32I-NEXT: sw a1, 0(a0) +; ALIGN-RV32I-NEXT: ret +; +; ALIGN-RV64I-LABEL: store_i64: +; ALIGN-RV64I: # %bb.0: +; ALIGN-RV64I-NEXT: sd a1, 0(a0) +; ALIGN-RV64I-NEXT: ret store i64 %v, i64* %p, align 1 ret void }