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 SISubtarget;
21 
22 class SIFrameLowering final : public AMDGPUFrameLowering {
23 public:
24   SIFrameLowering(StackDirection D, unsigned StackAl, int LAO,
25                   unsigned TransAl = 1) :
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 processFunctionBeforeFrameFinalized(
39     MachineFunction &MF,
40     RegScavenger *RS = nullptr) const override;
41 
42 private:
43   void emitFlatScratchInit(const SISubtarget &ST,
44                            MachineFunction &MF,
45                            MachineBasicBlock &MBB) const;
46 
47   unsigned getReservedPrivateSegmentBufferReg(
48     const SISubtarget &ST,
49     const SIInstrInfo *TII,
50     const SIRegisterInfo *TRI,
51     SIMachineFunctionInfo *MFI,
52     MachineFunction &MF) const;
53 
54   std::pair<unsigned, unsigned> getReservedPrivateSegmentWaveByteOffsetReg(
55     const SISubtarget &ST,
56     const SIInstrInfo *TII,
57     const SIRegisterInfo *TRI,
58     SIMachineFunctionInfo *MFI,
59     MachineFunction &MF) const;
60 
61   /// \brief Emits debugger prologue.
62   void emitDebuggerPrologue(MachineFunction &MF, MachineBasicBlock &MBB) const;
63 
64 public:
65   bool hasFP(const MachineFunction &MF) const override;
66   bool hasSP(const MachineFunction &MF) const;
67 };
68 
69 } // end namespace llvm
70 
71 #endif // LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
72