cc5a1b3dd9
MIR support is totally unusable for AMDGPU without this, since the set of reserved registers is set from fields here. Add a clone method to MachineFunctionInfo. This is a subtle variant of the copy constructor that is required if there are any MIR constructs that use pointers. Specifically, at minimum fields that reference MachineBasicBlocks or the MachineFunction need to be adjusted to the values in the new function.
117 lines
4.3 KiB
C++
117 lines
4.3 KiB
C++
//=- RISCVMachineFunctionInfo.h - RISCV machine function info -----*- C++ -*-=//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file declares RISCV-specific per-machine-function information.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_RISCV_RISCVMACHINEFUNCTIONINFO_H
|
|
#define LLVM_LIB_TARGET_RISCV_RISCVMACHINEFUNCTIONINFO_H
|
|
|
|
#include "RISCVSubtarget.h"
|
|
#include "llvm/CodeGen/MIRYamlMapping.h"
|
|
#include "llvm/CodeGen/MachineFrameInfo.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
|
|
namespace llvm {
|
|
|
|
class RISCVMachineFunctionInfo;
|
|
|
|
namespace yaml {
|
|
struct RISCVMachineFunctionInfo final : public yaml::MachineFunctionInfo {
|
|
int VarArgsFrameIndex;
|
|
int VarArgsSaveSize;
|
|
|
|
RISCVMachineFunctionInfo() = default;
|
|
RISCVMachineFunctionInfo(const llvm::RISCVMachineFunctionInfo &MFI);
|
|
|
|
void mappingImpl(yaml::IO &YamlIO) override;
|
|
~RISCVMachineFunctionInfo() = default;
|
|
};
|
|
|
|
template <> struct MappingTraits<RISCVMachineFunctionInfo> {
|
|
static void mapping(IO &YamlIO, RISCVMachineFunctionInfo &MFI) {
|
|
YamlIO.mapOptional("varArgsFrameIndex", MFI.VarArgsFrameIndex);
|
|
YamlIO.mapOptional("varArgsSaveSize", MFI.VarArgsSaveSize);
|
|
}
|
|
};
|
|
} // end namespace yaml
|
|
|
|
/// RISCVMachineFunctionInfo - This class is derived from MachineFunctionInfo
|
|
/// and contains private RISCV-specific information for each MachineFunction.
|
|
class RISCVMachineFunctionInfo : public MachineFunctionInfo {
|
|
private:
|
|
/// FrameIndex for start of varargs area
|
|
int VarArgsFrameIndex = 0;
|
|
/// Size of the save area used for varargs
|
|
int VarArgsSaveSize = 0;
|
|
/// FrameIndex used for transferring values between 64-bit FPRs and a pair
|
|
/// of 32-bit GPRs via the stack.
|
|
int MoveF64FrameIndex = -1;
|
|
/// Size of any opaque stack adjustment due to save/restore libcalls.
|
|
unsigned LibCallStackSize = 0;
|
|
/// Size of RVV stack.
|
|
uint64_t RVVStackSize = 0;
|
|
/// Alignment of RVV stack.
|
|
Align RVVStackAlign;
|
|
/// Padding required to keep RVV stack aligned within the main stack.
|
|
uint64_t RVVPadding = 0;
|
|
/// Size of stack frame to save callee saved registers
|
|
unsigned CalleeSavedStackSize = 0;
|
|
|
|
public:
|
|
RISCVMachineFunctionInfo(const MachineFunction &MF) {}
|
|
|
|
MachineFunctionInfo *
|
|
clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF,
|
|
const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB)
|
|
const override;
|
|
|
|
int getVarArgsFrameIndex() const { return VarArgsFrameIndex; }
|
|
void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; }
|
|
|
|
unsigned getVarArgsSaveSize() const { return VarArgsSaveSize; }
|
|
void setVarArgsSaveSize(int Size) { VarArgsSaveSize = Size; }
|
|
|
|
int getMoveF64FrameIndex(MachineFunction &MF) {
|
|
if (MoveF64FrameIndex == -1)
|
|
MoveF64FrameIndex =
|
|
MF.getFrameInfo().CreateStackObject(8, Align(8), false);
|
|
return MoveF64FrameIndex;
|
|
}
|
|
|
|
unsigned getLibCallStackSize() const { return LibCallStackSize; }
|
|
void setLibCallStackSize(unsigned Size) { LibCallStackSize = Size; }
|
|
|
|
bool useSaveRestoreLibCalls(const MachineFunction &MF) const {
|
|
// We cannot use fixed locations for the callee saved spill slots if the
|
|
// function uses a varargs save area, or is an interrupt handler.
|
|
return MF.getSubtarget<RISCVSubtarget>().enableSaveRestore() &&
|
|
VarArgsSaveSize == 0 && !MF.getFrameInfo().hasTailCall() &&
|
|
!MF.getFunction().hasFnAttribute("interrupt");
|
|
}
|
|
|
|
uint64_t getRVVStackSize() const { return RVVStackSize; }
|
|
void setRVVStackSize(uint64_t Size) { RVVStackSize = Size; }
|
|
|
|
Align getRVVStackAlign() const { return RVVStackAlign; }
|
|
void setRVVStackAlign(Align StackAlign) { RVVStackAlign = StackAlign; }
|
|
|
|
uint64_t getRVVPadding() const { return RVVPadding; }
|
|
void setRVVPadding(uint64_t Padding) { RVVPadding = Padding; }
|
|
|
|
unsigned getCalleeSavedStackSize() const { return CalleeSavedStackSize; }
|
|
void setCalleeSavedStackSize(unsigned Size) { CalleeSavedStackSize = Size; }
|
|
|
|
void initializeBaseYamlFields(const yaml::RISCVMachineFunctionInfo &YamlMFI);
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_LIB_TARGET_RISCV_RISCVMACHINEFUNCTIONINFO_H
|