1 //=- LoongArchMachineFunctionInfo.h - LoongArch machine function info -----===// 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 declares LoongArch-specific per-machine-function information. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #ifndef LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H 14 #define LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H 15 16 #include "LoongArchSubtarget.h" 17 #include "llvm/CodeGen/MachineFrameInfo.h" 18 #include "llvm/CodeGen/MachineFunction.h" 19 20 namespace llvm { 21 22 /// LoongArchMachineFunctionInfo - This class is derived from 23 /// MachineFunctionInfo and contains private LoongArch-specific information for 24 /// each MachineFunction. 25 class LoongArchMachineFunctionInfo : public MachineFunctionInfo { 26 private: 27 /// FrameIndex for start of varargs area 28 int VarArgsFrameIndex = 0; 29 /// Size of the save area used for varargs 30 int VarArgsSaveSize = 0; 31 32 /// Size of stack frame to save callee saved registers 33 unsigned CalleeSavedStackSize = 0; 34 35 /// FrameIndex of the spill slot when there is no scavenged register in 36 /// insertIndirectBranch. 37 int BranchRelaxationSpillFrameIndex = -1; 38 39 /// Registers that have been sign extended from i32. 40 SmallVector<Register, 8> SExt32Registers; 41 42 /// Pairs of `jr` instructions and corresponding JTI operands, used for the 43 /// `annotate-tablejump` option. 44 SmallVector<std::pair<MachineInstr *, MachineOperand *>, 4> JumpInfos; 45 46 public: 47 LoongArchMachineFunctionInfo(const Function &F, 48 const TargetSubtargetInfo *STI) {} 49 50 MachineFunctionInfo * 51 clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF, 52 const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB) 53 const override { 54 return DestMF.cloneInfo<LoongArchMachineFunctionInfo>(*this); 55 } 56 57 int getVarArgsFrameIndex() const { return VarArgsFrameIndex; } 58 void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; } 59 60 unsigned getVarArgsSaveSize() const { return VarArgsSaveSize; } 61 void setVarArgsSaveSize(int Size) { VarArgsSaveSize = Size; } 62 63 unsigned getCalleeSavedStackSize() const { return CalleeSavedStackSize; } 64 void setCalleeSavedStackSize(unsigned Size) { CalleeSavedStackSize = Size; } 65 66 int getBranchRelaxationSpillFrameIndex() { 67 return BranchRelaxationSpillFrameIndex; 68 } 69 void setBranchRelaxationSpillFrameIndex(int Index) { 70 BranchRelaxationSpillFrameIndex = Index; 71 } 72 73 void addSExt32Register(Register Reg) { SExt32Registers.push_back(Reg); } 74 75 bool isSExt32Register(Register Reg) const { 76 return is_contained(SExt32Registers, Reg); 77 } 78 79 void setJumpInfo(MachineInstr *JrMI, MachineOperand *JTIMO) { 80 JumpInfos.push_back(std::make_pair(JrMI, JTIMO)); 81 } 82 unsigned getJumpInfoSize() { return JumpInfos.size(); } 83 MachineInstr *getJumpInfoJrMI(unsigned Idx) { return JumpInfos[Idx].first; } 84 MachineOperand *getJumpInfoJTIMO(unsigned Idx) { 85 return JumpInfos[Idx].second; 86 } 87 }; 88 89 } // end namespace llvm 90 91 #endif // LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H 92