1 //===- SIMachineFunctionInfo.h - SIMachineFunctionInfo interface -*- C++ -*-==// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 /// \file 11 // 12 //===----------------------------------------------------------------------===// 13 14 15 #ifndef SIMACHINEFUNCTIONINFO_H_ 16 #define SIMACHINEFUNCTIONINFO_H_ 17 18 #include "AMDGPUMachineFunction.h" 19 #include <map> 20 21 namespace llvm { 22 23 class MachineRegisterInfo; 24 25 /// This class keeps track of the SPI_SP_INPUT_ADDR config register, which 26 /// tells the hardware which interpolation parameters to load. 27 class SIMachineFunctionInfo : public AMDGPUMachineFunction { 28 void anchor() override; 29 public: 30 31 struct SpilledReg { 32 unsigned VGPR; 33 int Lane; SpilledRegSpilledReg34 SpilledReg(unsigned R, int L) : VGPR (R), Lane (L) { } SpilledRegSpilledReg35 SpilledReg() : VGPR(0), Lane(-1) { } hasLaneSpilledReg36 bool hasLane() { return Lane != -1;} 37 }; 38 39 struct RegSpillTracker { 40 private: 41 unsigned CurrentLane; 42 std::map<unsigned, SpilledReg> SpilledRegisters; 43 public: 44 unsigned LaneVGPR; RegSpillTrackerRegSpillTracker45 RegSpillTracker() : CurrentLane(0), SpilledRegisters(), LaneVGPR(0) { } 46 /// \p NumRegs The number of consecutive registers what need to be spilled. 47 /// This function will ensure that all registers are stored in 48 /// the same VGPR. 49 /// \returns The lane to be used for storing the first register. 50 unsigned reserveLanes(MachineRegisterInfo &MRI, MachineFunction *MF, 51 unsigned NumRegs = 1); 52 void addSpilledReg(unsigned FrameIndex, unsigned Reg, int Lane = -1); 53 const SpilledReg& getSpilledReg(unsigned FrameIndex); programSpillsRegistersRegSpillTracker54 bool programSpillsRegisters() { return !SpilledRegisters.empty(); } 55 }; 56 57 // SIMachineFunctionInfo definition 58 59 SIMachineFunctionInfo(const MachineFunction &MF); 60 unsigned PSInputAddr; 61 struct RegSpillTracker SpillTracker; 62 }; 63 64 } // End namespace llvm 65 66 67 #endif //_SIMACHINEFUNCTIONINFO_H_ 68