1 //===--------------------- SIFrameLowering.h --------------------*- 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 #ifndef LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
11 #define LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
12 
13 #include "AMDGPUFrameLowering.h"
14 
15 namespace llvm {
16 
17 class SIInstrInfo;
18 class SIMachineFunctionInfo;
19 class SIRegisterInfo;
20 class GCNSubtarget;
21 
22 class SIFrameLowering final : public AMDGPUFrameLowering {
23 public:
24   SIFrameLowering(StackDirection D, unsigned StackAl, int LAO,
25                   unsigned TransAl = 1) :
AMDGPUFrameLowering(D,StackAl,LAO,TransAl)26     AMDGPUFrameLowering(D, StackAl, LAO, TransAl) {}
27   ~SIFrameLowering() override = default;
28 
29   void emitEntryFunctionPrologue(MachineFunction &MF,
30                                  MachineBasicBlock &MBB) const;
31   void emitPrologue(MachineFunction &MF,
32                     MachineBasicBlock &MBB) const override;
33   void emitEpilogue(MachineFunction &MF,
34                     MachineBasicBlock &MBB) const override;
35   int getFrameIndexReference(const MachineFunction &MF, int FI,
36                              unsigned &FrameReg) const override;
37 
38   void determineCalleeSaves(MachineFunction &MF, BitVector &SavedRegs,
39                             RegScavenger *RS = nullptr) const override;
40 
41   void processFunctionBeforeFrameFinalized(
42     MachineFunction &MF,
43     RegScavenger *RS = nullptr) const override;
44 
45   MachineBasicBlock::iterator
46   eliminateCallFramePseudoInstr(MachineFunction &MF,
47                                 MachineBasicBlock &MBB,
48                                 MachineBasicBlock::iterator MI) const override;
49 
50 private:
51   void emitFlatScratchInit(const GCNSubtarget &ST,
52                            MachineFunction &MF,
53                            MachineBasicBlock &MBB) const;
54 
55   unsigned getReservedPrivateSegmentBufferReg(
56     const GCNSubtarget &ST,
57     const SIInstrInfo *TII,
58     const SIRegisterInfo *TRI,
59     SIMachineFunctionInfo *MFI,
60     MachineFunction &MF) const;
61 
62   std::pair<unsigned, unsigned> getReservedPrivateSegmentWaveByteOffsetReg(
63     const GCNSubtarget &ST,
64     const SIInstrInfo *TII,
65     const SIRegisterInfo *TRI,
66     SIMachineFunctionInfo *MFI,
67     MachineFunction &MF) const;
68 
69   /// Emits debugger prologue.
70   void emitDebuggerPrologue(MachineFunction &MF, MachineBasicBlock &MBB) const;
71 
72   // Emit scratch setup code for AMDPAL or Mesa, assuming ResourceRegUsed is set.
73   void emitEntryFunctionScratchSetup(const GCNSubtarget &ST, MachineFunction &MF,
74       MachineBasicBlock &MBB, SIMachineFunctionInfo *MFI,
75       MachineBasicBlock::iterator I, unsigned PreloadedPrivateBufferReg,
76       unsigned ScratchRsrcReg) const;
77 
78 public:
79   bool hasFP(const MachineFunction &MF) const override;
80   bool hasSP(const MachineFunction &MF) const;
81 };
82 
83 } // end namespace llvm
84 
85 #endif // LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
86