1//===-- RISCVRegisterInfo.h - RISC-V Register Information Impl --*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file contains the RISC-V implementation of the TargetRegisterInfo class.
10//
11//===----------------------------------------------------------------------===//
12
13#ifndef LLVM_LIB_TARGET_RISCV_RISCVREGISTERINFO_H
14#define LLVM_LIB_TARGET_RISCV_RISCVREGISTERINFO_H
15
16#include "llvm/CodeGen/TargetRegisterInfo.h"
17#include "llvm/TargetParser/RISCVTargetParser.h"
18
19#define GET_REGINFO_HEADER
20#include "RISCVGenRegisterInfo.inc"
21
22namespace llvm {
23
24namespace RISCVRI {
25enum {
26 // The IsVRegClass value of this RegisterClass.
27 IsVRegClassShift = 0,
28 IsVRegClassShiftMask = 0b1 << IsVRegClassShift,
29 // The VLMul value of this RegisterClass. This value is valid iff IsVRegClass
30 // is true.
31 VLMulShift = IsVRegClassShift + 1,
32 VLMulShiftMask = 0b111 << VLMulShift,
33
34 // The NF value of this RegisterClass. This value is valid iff IsVRegClass is
35 // true.
36 NFShift = VLMulShift + 3,
37 NFShiftMask = 0b111 << NFShift,
38};
39
40/// \returns the IsVRegClass for the register class.
41static inline bool isVRegClass(uint64_t TSFlags) {
42 return TSFlags & IsVRegClassShiftMask >> IsVRegClassShift;
43}
44
45/// \returns the LMUL for the register class.
46static inline RISCVII::VLMUL getLMul(uint64_t TSFlags) {
47 return static_cast<RISCVII::VLMUL>((TSFlags & VLMulShiftMask) >> VLMulShift);
48}
49
50/// \returns the NF for the register class.
51static inline unsigned getNF(uint64_t TSFlags) {
52 return static_cast<unsigned>((TSFlags & NFShiftMask) >> NFShift) + 1;
53}
54} // namespace RISCVRI
55
56struct RISCVRegisterInfo : public RISCVGenRegisterInfo {
57
58 RISCVRegisterInfo(unsigned HwMode);
59
60 const uint32_t *getCallPreservedMask(const MachineFunction &MF,
61 CallingConv::ID) const override;
62
63 const MCPhysReg *getCalleeSavedRegs(const MachineFunction *MF) const override;
64
65 BitVector getReservedRegs(const MachineFunction &MF) const override;
66 bool isAsmClobberable(const MachineFunction &MF,
67 MCRegister PhysReg) const override;
68
69 const uint32_t *getNoPreservedMask() const override;
70
71 // Update DestReg to have the value SrcReg plus an offset. This is
72 // used during frame layout, and we may need to ensure that if we
73 // split the offset internally that the DestReg is always aligned,
74 // assuming that source reg was.
75 void adjustReg(MachineBasicBlock &MBB, MachineBasicBlock::iterator II,
76 const DebugLoc &DL, Register DestReg, Register SrcReg,
77 StackOffset Offset, MachineInstr::MIFlag Flag,
78 MaybeAlign RequiredAlign) const;
79
80 bool eliminateFrameIndex(MachineBasicBlock::iterator MI, int SPAdj,
81 unsigned FIOperandNum,
82 RegScavenger *RS = nullptr) const override;
83
84 bool requiresVirtualBaseRegisters(const MachineFunction &MF) const override;
85
86 bool needsFrameBaseReg(MachineInstr *MI, int64_t Offset) const override;
87
88 bool isFrameOffsetLegal(const MachineInstr *MI, Register BaseReg,
89 int64_t Offset) const override;
90
91 Register materializeFrameBaseRegister(MachineBasicBlock *MBB, int FrameIdx,
92 int64_t Offset) const override;
93
94 void resolveFrameIndex(MachineInstr &MI, Register BaseReg,
95 int64_t Offset) const override;
96
97 int64_t getFrameIndexInstrOffset(const MachineInstr *MI,
98 int Idx) const override;
99
100 void lowerVSPILL(MachineBasicBlock::iterator II) const;
101 void lowerVRELOAD(MachineBasicBlock::iterator II) const;
102
103 Register getFrameRegister(const MachineFunction &MF) const override;
104
105 bool requiresRegisterScavenging(const MachineFunction &MF) const override {
106 return true;
107 }
108
109 bool requiresFrameIndexScavenging(const MachineFunction &MF) const override {
110 return true;
111 }
112
113 const TargetRegisterClass *
114 getPointerRegClass(const MachineFunction &MF,
115 unsigned Kind = 0) const override {
116 return &RISCV::GPRRegClass;
117 }
118
119 const TargetRegisterClass *
120 getLargestLegalSuperClass(const TargetRegisterClass *RC,
121 const MachineFunction &) const override;
122
123 void getOffsetOpcodes(const StackOffset &Offset,
124 SmallVectorImpl<uint64_t> &Ops) const override;
125
126 unsigned getRegisterCostTableIndex(const MachineFunction &MF) const override;
127
128 bool getRegAllocationHints(Register VirtReg, ArrayRef<MCPhysReg> Order,
129 SmallVectorImpl<MCPhysReg> &Hints,
130 const MachineFunction &MF, const VirtRegMap *VRM,
131 const LiveRegMatrix *Matrix) const override;
132
133 const TargetRegisterClass *
134 getLargestSuperClass(const TargetRegisterClass *RC) const override {
135 if (RISCV::VRM8RegClass.hasSubClassEq(RC))
136 return &RISCV::VRM8RegClass;
137 if (RISCV::VRM4RegClass.hasSubClassEq(RC))
138 return &RISCV::VRM4RegClass;
139 if (RISCV::VRM2RegClass.hasSubClassEq(RC))
140 return &RISCV::VRM2RegClass;
141 if (RISCV::VRRegClass.hasSubClassEq(RC))
142 return &RISCV::VRRegClass;
143 return RC;
144 }
145
146 bool doesRegClassHavePseudoInitUndef(
147 const TargetRegisterClass *RC) const override {
148 return isVRRegClass(RC);
149 }
150
151 static bool isVRRegClass(const TargetRegisterClass *RC) {
152 return RISCVRI::isVRegClass(TSFlags: RC->TSFlags) &&
153 RISCVRI::getNF(TSFlags: RC->TSFlags) == 1;
154 }
155
156 static bool isVRNRegClass(const TargetRegisterClass *RC) {
157 return RISCVRI::isVRegClass(TSFlags: RC->TSFlags) && RISCVRI::getNF(TSFlags: RC->TSFlags) > 1;
158 }
159
160 static bool isRVVRegClass(const TargetRegisterClass *RC) {
161 return RISCVRI::isVRegClass(TSFlags: RC->TSFlags);
162 }
163};
164} // namespace llvm
165
166#endif
167