1 //===--------------------- SIFrameLowering.h --------------------*- 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 #ifndef LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
10 #define LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
11 
12 #include "AMDGPUFrameLowering.h"
13 
14 namespace llvm {
15 
16 class SIInstrInfo;
17 class SIMachineFunctionInfo;
18 class SIRegisterInfo;
19 class GCNSubtarget;
20 
21 class SIFrameLowering final : public AMDGPUFrameLowering {
22 public:
23   SIFrameLowering(StackDirection D, Align StackAl, int LAO,
24                   Align TransAl = Align(1))
AMDGPUFrameLowering(D,StackAl,LAO,TransAl)25       : AMDGPUFrameLowering(D, StackAl, LAO, TransAl) {}
26   ~SIFrameLowering() override = default;
27 
28   void emitEntryFunctionPrologue(MachineFunction &MF,
29                                  MachineBasicBlock &MBB) const;
30   void emitPrologue(MachineFunction &MF,
31                     MachineBasicBlock &MBB) const override;
32   void emitEpilogue(MachineFunction &MF,
33                     MachineBasicBlock &MBB) const override;
34   StackOffset getFrameIndexReference(const MachineFunction &MF, int FI,
35                                      Register &FrameReg) const override;
36 
37   void determineCalleeSaves(MachineFunction &MF, BitVector &SavedRegs,
38                             RegScavenger *RS = nullptr) const override;
39   void determineCalleeSavesSGPR(MachineFunction &MF, BitVector &SavedRegs,
40                                 RegScavenger *RS = nullptr) const;
41   bool
42   assignCalleeSavedSpillSlots(MachineFunction &MF,
43                               const TargetRegisterInfo *TRI,
44                               std::vector<CalleeSavedInfo> &CSI) const override;
45 
46   bool isSupportedStackID(TargetStackID::Value ID) const override;
47 
48   void processFunctionBeforeFrameFinalized(
49     MachineFunction &MF,
50     RegScavenger *RS = nullptr) const override;
51 
52   MachineBasicBlock::iterator
53   eliminateCallFramePseudoInstr(MachineFunction &MF,
54                                 MachineBasicBlock &MBB,
55                                 MachineBasicBlock::iterator MI) const override;
56 
57 private:
58   void emitEntryFunctionFlatScratchInit(MachineFunction &MF,
59                                         MachineBasicBlock &MBB,
60                                         MachineBasicBlock::iterator I,
61                                         const DebugLoc &DL,
62                                         Register ScratchWaveOffsetReg) const;
63 
64   Register getEntryFunctionReservedScratchRsrcReg(MachineFunction &MF) const;
65 
66   void emitEntryFunctionScratchRsrcRegSetup(
67       MachineFunction &MF, MachineBasicBlock &MBB,
68       MachineBasicBlock::iterator I, const DebugLoc &DL,
69       Register PreloadedPrivateBufferReg, Register ScratchRsrcReg,
70       Register ScratchWaveOffsetReg) const;
71 
72 public:
73   bool hasFP(const MachineFunction &MF) const override;
74 
75   bool requiresStackPointerReference(const MachineFunction &MF) const;
76 };
77 
78 } // end namespace llvm
79 
80 #endif // LLVM_LIB_TARGET_AMDGPU_SIFRAMELOWERING_H
81