1 //=- RISCVMachineFunctionInfo.h - RISCV machine function info -----*- 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 // This file declares RISCV-specific per-machine-function information.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef LLVM_LIB_TARGET_RISCV_RISCVMACHINEFUNCTIONINFO_H
14 #define LLVM_LIB_TARGET_RISCV_RISCVMACHINEFUNCTIONINFO_H
15 
16 #include "RISCVSubtarget.h"
17 #include "llvm/CodeGen/MachineFrameInfo.h"
18 #include "llvm/CodeGen/MachineFunction.h"
19 
20 namespace llvm {
21 
22 /// RISCVMachineFunctionInfo - This class is derived from MachineFunctionInfo
23 /// and contains private RISCV-specific information for each MachineFunction.
24 class RISCVMachineFunctionInfo : public MachineFunctionInfo {
25 private:
26   /// FrameIndex for start of varargs area
27   int VarArgsFrameIndex = 0;
28   /// Size of the save area used for varargs
29   int VarArgsSaveSize = 0;
30   /// FrameIndex used for transferring values between 64-bit FPRs and a pair
31   /// of 32-bit GPRs via the stack.
32   int MoveF64FrameIndex = -1;
33   /// Size of any opaque stack adjustment due to save/restore libcalls.
34   unsigned LibCallStackSize = 0;
35   /// Size of RVV stack.
36   uint64_t RVVStackSize = 0;
37   /// Padding required to keep RVV stack aligned within the main stack.
38   uint64_t RVVPadding = 0;
39   /// Size of stack frame to save callee saved registers
40   unsigned CalleeSavedStackSize = 0;
41 
42 public:
43   RISCVMachineFunctionInfo(const MachineFunction &MF) {}
44 
45   int getVarArgsFrameIndex() const { return VarArgsFrameIndex; }
46   void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; }
47 
48   unsigned getVarArgsSaveSize() const { return VarArgsSaveSize; }
49   void setVarArgsSaveSize(int Size) { VarArgsSaveSize = Size; }
50 
51   int getMoveF64FrameIndex(MachineFunction &MF) {
52     if (MoveF64FrameIndex == -1)
53       MoveF64FrameIndex =
54           MF.getFrameInfo().CreateStackObject(8, Align(8), false);
55     return MoveF64FrameIndex;
56   }
57 
58   unsigned getLibCallStackSize() const { return LibCallStackSize; }
59   void setLibCallStackSize(unsigned Size) { LibCallStackSize = Size; }
60 
61   bool useSaveRestoreLibCalls(const MachineFunction &MF) const {
62     // We cannot use fixed locations for the callee saved spill slots if the
63     // function uses a varargs save area, or is an interrupt handler.
64     return MF.getSubtarget<RISCVSubtarget>().enableSaveRestore() &&
65            VarArgsSaveSize == 0 && !MF.getFrameInfo().hasTailCall() &&
66            !MF.getFunction().hasFnAttribute("interrupt");
67   }
68 
69   uint64_t getRVVStackSize() const { return RVVStackSize; }
70   void setRVVStackSize(uint64_t Size) { RVVStackSize = Size; }
71 
72   uint64_t getRVVPadding() const { return RVVPadding; }
73   void setRVVPadding(uint64_t Padding) { RVVPadding = Padding; }
74 
75   unsigned getCalleeSavedStackSize() const { return CalleeSavedStackSize; }
76   void setCalleeSavedStackSize(unsigned Size) { CalleeSavedStackSize = Size; }
77 };
78 
79 } // end namespace llvm
80 
81 #endif // LLVM_LIB_TARGET_RISCV_RISCVMACHINEFUNCTIONINFO_H
82