1 //===-- RISCVRegisterInfo.cpp - RISCV Register Information ------*- 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 RISCV implementation of the TargetRegisterInfo class.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "RISCVRegisterInfo.h"
14 #include "RISCV.h"
15 #include "RISCVMachineFunctionInfo.h"
16 #include "RISCVSubtarget.h"
17 #include "llvm/CodeGen/MachineFrameInfo.h"
18 #include "llvm/CodeGen/MachineFunction.h"
19 #include "llvm/CodeGen/MachineInstrBuilder.h"
20 #include "llvm/CodeGen/RegisterScavenging.h"
21 #include "llvm/CodeGen/TargetFrameLowering.h"
22 #include "llvm/CodeGen/TargetInstrInfo.h"
23 #include "llvm/Support/ErrorHandling.h"
24 
25 #define GET_REGINFO_TARGET_DESC
26 #include "RISCVGenRegisterInfo.inc"
27 
28 using namespace llvm;
29 
30 static_assert(RISCV::X1 == RISCV::X0 + 1, "Register list not consecutive");
31 static_assert(RISCV::X31 == RISCV::X0 + 31, "Register list not consecutive");
32 static_assert(RISCV::F1_H == RISCV::F0_H + 1, "Register list not consecutive");
33 static_assert(RISCV::F31_H == RISCV::F0_H + 31,
34               "Register list not consecutive");
35 static_assert(RISCV::F1_F == RISCV::F0_F + 1, "Register list not consecutive");
36 static_assert(RISCV::F31_F == RISCV::F0_F + 31,
37               "Register list not consecutive");
38 static_assert(RISCV::F1_D == RISCV::F0_D + 1, "Register list not consecutive");
39 static_assert(RISCV::F31_D == RISCV::F0_D + 31,
40               "Register list not consecutive");
41 static_assert(RISCV::V1 == RISCV::V0 + 1, "Register list not consecutive");
42 static_assert(RISCV::V31 == RISCV::V0 + 31, "Register list not consecutive");
43 
44 RISCVRegisterInfo::RISCVRegisterInfo(unsigned HwMode)
45     : RISCVGenRegisterInfo(RISCV::X1, /*DwarfFlavour*/0, /*EHFlavor*/0,
46                            /*PC*/0, HwMode) {}
47 
48 const MCPhysReg *
49 RISCVRegisterInfo::getCalleeSavedRegs(const MachineFunction *MF) const {
50   auto &Subtarget = MF->getSubtarget<RISCVSubtarget>();
51   if (MF->getFunction().getCallingConv() == CallingConv::GHC)
52     return CSR_NoRegs_SaveList;
53   if (MF->getFunction().hasFnAttribute("interrupt")) {
54     if (Subtarget.hasStdExtD())
55       return CSR_XLEN_F64_Interrupt_SaveList;
56     if (Subtarget.hasStdExtF())
57       return CSR_XLEN_F32_Interrupt_SaveList;
58     return CSR_Interrupt_SaveList;
59   }
60 
61   switch (Subtarget.getTargetABI()) {
62   default:
63     llvm_unreachable("Unrecognized ABI");
64   case RISCVABI::ABI_ILP32:
65   case RISCVABI::ABI_LP64:
66     return CSR_ILP32_LP64_SaveList;
67   case RISCVABI::ABI_ILP32F:
68   case RISCVABI::ABI_LP64F:
69     return CSR_ILP32F_LP64F_SaveList;
70   case RISCVABI::ABI_ILP32D:
71   case RISCVABI::ABI_LP64D:
72     return CSR_ILP32D_LP64D_SaveList;
73   }
74 }
75 
76 BitVector RISCVRegisterInfo::getReservedRegs(const MachineFunction &MF) const {
77   const RISCVFrameLowering *TFI = getFrameLowering(MF);
78   BitVector Reserved(getNumRegs());
79 
80   // Mark any registers requested to be reserved as such
81   for (size_t Reg = 0; Reg < getNumRegs(); Reg++) {
82     if (MF.getSubtarget<RISCVSubtarget>().isRegisterReservedByUser(Reg))
83       markSuperRegs(Reserved, Reg);
84   }
85 
86   // Use markSuperRegs to ensure any register aliases are also reserved
87   markSuperRegs(Reserved, RISCV::X0); // zero
88   markSuperRegs(Reserved, RISCV::X2); // sp
89   markSuperRegs(Reserved, RISCV::X3); // gp
90   markSuperRegs(Reserved, RISCV::X4); // tp
91   if (TFI->hasFP(MF))
92     markSuperRegs(Reserved, RISCV::X8); // fp
93   // Reserve the base register if we need to realign the stack and allocate
94   // variable-sized objects at runtime.
95   if (TFI->hasBP(MF))
96     markSuperRegs(Reserved, RISCVABI::getBPReg()); // bp
97 
98   // V registers for code generation. We handle them manually.
99   markSuperRegs(Reserved, RISCV::VL);
100   markSuperRegs(Reserved, RISCV::VTYPE);
101   markSuperRegs(Reserved, RISCV::VXSAT);
102   markSuperRegs(Reserved, RISCV::VXRM);
103 
104   // Floating point environment registers.
105   markSuperRegs(Reserved, RISCV::FRM);
106   markSuperRegs(Reserved, RISCV::FFLAGS);
107   markSuperRegs(Reserved, RISCV::FCSR);
108 
109   assert(checkAllSuperRegsMarked(Reserved));
110   return Reserved;
111 }
112 
113 bool RISCVRegisterInfo::isAsmClobberable(const MachineFunction &MF,
114                                          MCRegister PhysReg) const {
115   return !MF.getSubtarget<RISCVSubtarget>().isRegisterReservedByUser(PhysReg);
116 }
117 
118 bool RISCVRegisterInfo::isConstantPhysReg(MCRegister PhysReg) const {
119   return PhysReg == RISCV::X0;
120 }
121 
122 const uint32_t *RISCVRegisterInfo::getNoPreservedMask() const {
123   return CSR_NoRegs_RegMask;
124 }
125 
126 // Frame indexes representing locations of CSRs which are given a fixed location
127 // by save/restore libcalls.
128 static const std::map<unsigned, int> FixedCSRFIMap = {
129   {/*ra*/  RISCV::X1,   -1},
130   {/*s0*/  RISCV::X8,   -2},
131   {/*s1*/  RISCV::X9,   -3},
132   {/*s2*/  RISCV::X18,  -4},
133   {/*s3*/  RISCV::X19,  -5},
134   {/*s4*/  RISCV::X20,  -6},
135   {/*s5*/  RISCV::X21,  -7},
136   {/*s6*/  RISCV::X22,  -8},
137   {/*s7*/  RISCV::X23,  -9},
138   {/*s8*/  RISCV::X24,  -10},
139   {/*s9*/  RISCV::X25,  -11},
140   {/*s10*/ RISCV::X26,  -12},
141   {/*s11*/ RISCV::X27,  -13}
142 };
143 
144 bool RISCVRegisterInfo::hasReservedSpillSlot(const MachineFunction &MF,
145                                              Register Reg,
146                                              int &FrameIdx) const {
147   const auto *RVFI = MF.getInfo<RISCVMachineFunctionInfo>();
148   if (!RVFI->useSaveRestoreLibCalls(MF))
149     return false;
150 
151   auto FII = FixedCSRFIMap.find(Reg);
152   if (FII == FixedCSRFIMap.end())
153     return false;
154 
155   FrameIdx = FII->second;
156   return true;
157 }
158 
159 static bool isRVVWholeLoadStore(unsigned Opcode) {
160   switch (Opcode) {
161   default:
162     return false;
163   case RISCV::VS1R_V:
164   case RISCV::VS2R_V:
165   case RISCV::VS4R_V:
166   case RISCV::VS8R_V:
167   case RISCV::VL1RE8_V:
168   case RISCV::VL2RE8_V:
169   case RISCV::VL4RE8_V:
170   case RISCV::VL8RE8_V:
171   case RISCV::VL1RE16_V:
172   case RISCV::VL2RE16_V:
173   case RISCV::VL4RE16_V:
174   case RISCV::VL8RE16_V:
175   case RISCV::VL1RE32_V:
176   case RISCV::VL2RE32_V:
177   case RISCV::VL4RE32_V:
178   case RISCV::VL8RE32_V:
179   case RISCV::VL1RE64_V:
180   case RISCV::VL2RE64_V:
181   case RISCV::VL4RE64_V:
182   case RISCV::VL8RE64_V:
183     return true;
184   }
185 }
186 
187 void RISCVRegisterInfo::eliminateFrameIndex(MachineBasicBlock::iterator II,
188                                             int SPAdj, unsigned FIOperandNum,
189                                             RegScavenger *RS) const {
190   assert(SPAdj == 0 && "Unexpected non-zero SPAdj value");
191 
192   MachineInstr &MI = *II;
193   MachineFunction &MF = *MI.getParent()->getParent();
194   MachineRegisterInfo &MRI = MF.getRegInfo();
195   const RISCVInstrInfo *TII = MF.getSubtarget<RISCVSubtarget>().getInstrInfo();
196   DebugLoc DL = MI.getDebugLoc();
197 
198   int FrameIndex = MI.getOperand(FIOperandNum).getIndex();
199   Register FrameReg;
200   StackOffset Offset =
201       getFrameLowering(MF)->getFrameIndexReference(MF, FrameIndex, FrameReg);
202   bool isRVV = RISCVVPseudosTable::getPseudoInfo(MI.getOpcode()) ||
203                isRVVWholeLoadStore(MI.getOpcode()) ||
204                TII->isRVVSpillForZvlsseg(MI.getOpcode());
205   if (!isRVV)
206     Offset += StackOffset::getFixed(MI.getOperand(FIOperandNum + 1).getImm());
207 
208   if (!isInt<32>(Offset.getFixed())) {
209     report_fatal_error(
210         "Frame offsets outside of the signed 32-bit range not supported");
211   }
212 
213   MachineBasicBlock &MBB = *MI.getParent();
214   bool FrameRegIsKill = false;
215 
216   // If required, pre-compute the scalable factor amount which will be used in
217   // later offset computation. Since this sequence requires up to two scratch
218   // registers -- after which one is made free -- this grants us better
219   // scavenging of scratch registers as only up to two are live at one time,
220   // rather than three.
221   Register ScalableFactorRegister;
222   unsigned ScalableAdjOpc = RISCV::ADD;
223   if (Offset.getScalable()) {
224     int64_t ScalableValue = Offset.getScalable();
225     if (ScalableValue < 0) {
226       ScalableValue = -ScalableValue;
227       ScalableAdjOpc = RISCV::SUB;
228     }
229     // 1. Get vlenb && multiply vlen with the number of vector registers.
230     ScalableFactorRegister =
231         TII->getVLENFactoredAmount(MF, MBB, II, ScalableValue);
232   }
233 
234   if (!isInt<12>(Offset.getFixed())) {
235     // The offset won't fit in an immediate, so use a scratch register instead
236     // Modify Offset and FrameReg appropriately
237     Register ScratchReg = MRI.createVirtualRegister(&RISCV::GPRRegClass);
238     TII->movImm(MBB, II, DL, ScratchReg, Offset.getFixed());
239     if (MI.getOpcode() == RISCV::ADDI && !Offset.getScalable()) {
240       BuildMI(MBB, II, DL, TII->get(RISCV::ADD), MI.getOperand(0).getReg())
241         .addReg(FrameReg)
242         .addReg(ScratchReg, RegState::Kill);
243       MI.eraseFromParent();
244       return;
245     }
246     BuildMI(MBB, II, DL, TII->get(RISCV::ADD), ScratchReg)
247         .addReg(FrameReg)
248         .addReg(ScratchReg, RegState::Kill);
249     Offset = StackOffset::get(0, Offset.getScalable());
250     FrameReg = ScratchReg;
251     FrameRegIsKill = true;
252   }
253 
254   if (!Offset.getScalable()) {
255     // Offset = (fixed offset, 0)
256     MI.getOperand(FIOperandNum)
257         .ChangeToRegister(FrameReg, false, false, FrameRegIsKill);
258     if (!isRVV)
259       MI.getOperand(FIOperandNum + 1).ChangeToImmediate(Offset.getFixed());
260     else {
261       if (Offset.getFixed()) {
262         Register ScratchReg = MRI.createVirtualRegister(&RISCV::GPRRegClass);
263         BuildMI(MBB, II, DL, TII->get(RISCV::ADDI), ScratchReg)
264           .addReg(FrameReg, getKillRegState(FrameRegIsKill))
265           .addImm(Offset.getFixed());
266         MI.getOperand(FIOperandNum)
267           .ChangeToRegister(ScratchReg, false, false, true);
268       }
269     }
270   } else {
271     // Offset = (fixed offset, scalable offset)
272     // Step 1, the scalable offset, has already been computed.
273     assert(ScalableFactorRegister &&
274            "Expected pre-computation of scalable factor in earlier step");
275 
276     // 2. Calculate address: FrameReg + result of multiply
277     if (MI.getOpcode() == RISCV::ADDI && !Offset.getFixed()) {
278       BuildMI(MBB, II, DL, TII->get(ScalableAdjOpc), MI.getOperand(0).getReg())
279           .addReg(FrameReg, getKillRegState(FrameRegIsKill))
280           .addReg(ScalableFactorRegister, RegState::Kill);
281       MI.eraseFromParent();
282       return;
283     }
284     Register VL = MRI.createVirtualRegister(&RISCV::GPRRegClass);
285     BuildMI(MBB, II, DL, TII->get(ScalableAdjOpc), VL)
286         .addReg(FrameReg, getKillRegState(FrameRegIsKill))
287         .addReg(ScalableFactorRegister, RegState::Kill);
288 
289     if (isRVV && Offset.getFixed()) {
290       // Scalable load/store has no immediate argument. We need to add the
291       // fixed part into the load/store base address.
292       BuildMI(MBB, II, DL, TII->get(RISCV::ADDI), VL)
293           .addReg(VL)
294           .addImm(Offset.getFixed());
295     }
296 
297     // 3. Replace address register with calculated address register
298     MI.getOperand(FIOperandNum).ChangeToRegister(VL, false, false, true);
299     if (!isRVV)
300       MI.getOperand(FIOperandNum + 1).ChangeToImmediate(Offset.getFixed());
301   }
302 
303   MachineFrameInfo &MFI = MF.getFrameInfo();
304   auto ZvlssegInfo = TII->isRVVSpillForZvlsseg(MI.getOpcode());
305   if (ZvlssegInfo) {
306     int64_t ScalableValue = MFI.getObjectSize(FrameIndex) / ZvlssegInfo->first;
307     Register FactorRegister =
308         TII->getVLENFactoredAmount(MF, MBB, II, ScalableValue);
309     MI.getOperand(FIOperandNum + 1)
310         .ChangeToRegister(FactorRegister, /*isDef=*/false);
311   }
312 }
313 
314 Register RISCVRegisterInfo::getFrameRegister(const MachineFunction &MF) const {
315   const TargetFrameLowering *TFI = getFrameLowering(MF);
316   return TFI->hasFP(MF) ? RISCV::X8 : RISCV::X2;
317 }
318 
319 const uint32_t *
320 RISCVRegisterInfo::getCallPreservedMask(const MachineFunction & MF,
321                                         CallingConv::ID CC) const {
322   auto &Subtarget = MF.getSubtarget<RISCVSubtarget>();
323 
324   if (CC == CallingConv::GHC)
325     return CSR_NoRegs_RegMask;
326   switch (Subtarget.getTargetABI()) {
327   default:
328     llvm_unreachable("Unrecognized ABI");
329   case RISCVABI::ABI_ILP32:
330   case RISCVABI::ABI_LP64:
331     return CSR_ILP32_LP64_RegMask;
332   case RISCVABI::ABI_ILP32F:
333   case RISCVABI::ABI_LP64F:
334     return CSR_ILP32F_LP64F_RegMask;
335   case RISCVABI::ABI_ILP32D:
336   case RISCVABI::ABI_LP64D:
337     return CSR_ILP32D_LP64D_RegMask;
338   }
339 }
340 
341 const TargetRegisterClass *
342 RISCVRegisterInfo::getLargestLegalSuperClass(const TargetRegisterClass *RC,
343                                              const MachineFunction &) const {
344   if (RC == &RISCV::VMV0RegClass)
345     return &RISCV::VRRegClass;
346   return RC;
347 }
348