Home | History | Annotate | Line # | Download | only in RISCV 
      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