[llvm] 98798a5 - [RISCV] Add helper function for RVV intrinsics in getTgtMemIntrinsic. NFC
Craig Topper via llvm-commits
llvm-commits at lists.llvm.org
Wed Mar 29 15:34:34 PDT 2023
Author: Craig Topper
Date: 2023-03-29T15:34:25-07:00
New Revision: 98798a5a900ed8a327499fc33a2e89a45730df89
URL: https://github.com/llvm/llvm-project/commit/98798a5a900ed8a327499fc33a2e89a45730df89
DIFF: https://github.com/llvm/llvm-project/commit/98798a5a900ed8a327499fc33a2e89a45730df89.diff
LOG: [RISCV] Add helper function for RVV intrinsics in getTgtMemIntrinsic. NFC
Preparation for adding the other RVV load/store intrinsics we use
for the C API.
Reviewed By: asb, kito-cheng
Differential Revision: https://reviews.llvm.org/D147004
Added:
Modified:
llvm/lib/Target/RISCV/RISCVISelLowering.cpp
Removed:
################################################################################
diff --git a/llvm/lib/Target/RISCV/RISCVISelLowering.cpp b/llvm/lib/Target/RISCV/RISCVISelLowering.cpp
index dd371d4bcac8..b3808cbd9782 100644
--- a/llvm/lib/Target/RISCV/RISCVISelLowering.cpp
+++ b/llvm/lib/Target/RISCV/RISCVISelLowering.cpp
@@ -1133,6 +1133,36 @@ bool RISCVTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info,
MachineFunction &MF,
unsigned Intrinsic) const {
auto &DL = I.getModule()->getDataLayout();
+
+ auto SetRVVLoadStoreInfo = [&](unsigned PtrOp, bool IsStore,
+ bool IsUnitStrided) {
+ Info.opc = IsStore ? ISD::INTRINSIC_VOID : ISD::INTRINSIC_W_CHAIN;
+ Info.ptrVal = I.getArgOperand(PtrOp);
+ Type *MemTy;
+ if (IsStore) {
+ // Store value is the first operand.
+ MemTy = I.getArgOperand(0)->getType();
+ } else {
+ // Use return type. If it's segment load, return type is a struct.
+ MemTy = I.getType();
+ if (MemTy->isStructTy())
+ MemTy = MemTy->getStructElementType(0);
+ }
+ if (!IsUnitStrided)
+ MemTy = MemTy->getScalarType();
+
+ Info.memVT = getValueType(DL, MemTy);
+ Info.align = Align(DL.getTypeSizeInBits(MemTy->getScalarType()) / 8);
+ Info.size = MemoryLocation::UnknownSize;
+ Info.flags |=
+ IsStore ? MachineMemOperand::MOStore : MachineMemOperand::MOLoad;
+ return true;
+ };
+
+ if (I.getMetadata(LLVMContext::MD_nontemporal) != nullptr)
+ Info.flags |= MachineMemOperand::MONonTemporal;
+
+ Info.flags |= RISCVTargetLowering::getTargetMMOFlags(I);
switch (Intrinsic) {
default:
return false;
@@ -1154,24 +1184,11 @@ bool RISCVTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info,
MachineMemOperand::MOVolatile;
return true;
case Intrinsic::riscv_masked_strided_load:
- Info.opc = ISD::INTRINSIC_W_CHAIN;
- Info.ptrVal = I.getArgOperand(1);
- Info.memVT = getValueType(DL, I.getType()->getScalarType());
- Info.align = Align(DL.getTypeSizeInBits(I.getType()->getScalarType()) / 8);
- Info.size = MemoryLocation::UnknownSize;
- Info.flags |= MachineMemOperand::MOLoad;
- return true;
+ return SetRVVLoadStoreInfo(/*PtrOp*/ 1, /*IsStore*/ false,
+ /*IsUnitStrided*/ false);
case Intrinsic::riscv_masked_strided_store:
- Info.opc = ISD::INTRINSIC_VOID;
- Info.ptrVal = I.getArgOperand(1);
- Info.memVT =
- getValueType(DL, I.getArgOperand(0)->getType()->getScalarType());
- Info.align = Align(
- DL.getTypeSizeInBits(I.getArgOperand(0)->getType()->getScalarType()) /
- 8);
- Info.size = MemoryLocation::UnknownSize;
- Info.flags |= MachineMemOperand::MOStore;
- return true;
+ return SetRVVLoadStoreInfo(/*PtrOp*/ 1, /*IsStore*/ true,
+ /*IsUnitStrided*/ false);
case Intrinsic::riscv_seg2_load:
case Intrinsic::riscv_seg3_load:
case Intrinsic::riscv_seg4_load:
@@ -1179,17 +1196,8 @@ bool RISCVTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info,
case Intrinsic::riscv_seg6_load:
case Intrinsic::riscv_seg7_load:
case Intrinsic::riscv_seg8_load:
- Info.opc = ISD::INTRINSIC_W_CHAIN;
- Info.ptrVal = I.getArgOperand(0);
- Info.memVT =
- getValueType(DL, I.getType()->getStructElementType(0)->getScalarType());
- Info.align =
- Align(DL.getTypeSizeInBits(
- I.getType()->getStructElementType(0)->getScalarType()) /
- 8);
- Info.size = MemoryLocation::UnknownSize;
- Info.flags |= MachineMemOperand::MOLoad;
- return true;
+ return SetRVVLoadStoreInfo(/*PtrOp*/ 0, /*IsStore*/ false,
+ /*IsUnitStrided*/ false);
case Intrinsic::riscv_seg2_store:
case Intrinsic::riscv_seg3_store:
case Intrinsic::riscv_seg4_store:
@@ -1197,17 +1205,10 @@ bool RISCVTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info,
case Intrinsic::riscv_seg6_store:
case Intrinsic::riscv_seg7_store:
case Intrinsic::riscv_seg8_store:
- Info.opc = ISD::INTRINSIC_VOID;
// Operands are (vec, ..., vec, ptr, vl, int_id)
- Info.ptrVal = I.getArgOperand(I.getNumOperands() - 3);
- Info.memVT =
- getValueType(DL, I.getArgOperand(0)->getType()->getScalarType());
- Info.align = Align(
- DL.getTypeSizeInBits(I.getArgOperand(0)->getType()->getScalarType()) /
- 8);
- Info.size = MemoryLocation::UnknownSize;
- Info.flags |= MachineMemOperand::MOStore;
- return true;
+ return SetRVVLoadStoreInfo(/*PtrOp*/ I.getNumOperands() - 3,
+ /*IsStore*/ true,
+ /*IsUnitStrided*/ false);
}
}
More information about the llvm-commits
mailing list