llvm-project/llvm/lib/Target/AVR/AVRMachineFunctionInfo.h
Matt Arsenault 69e75ae695 CodeGen: Don't lazily construct MachineFunctionInfo
This fixes what I consider to be an API flaw I've tripped over
multiple times. The point this is constructed isn't well defined, so
depending on where this is first called, you can conclude different
information based on the MachineFunction. For example, the AMDGPU
implementation inspected the MachineFrameInfo on construction for the
stack objects and if the frame has calls. This kind of worked in
SelectionDAG which visited all allocas up front, but broke in
GlobalISel which hasn't visited any of the IR when arguments are
lowered.

I've run into similar problems before with the MIR parser and trying
to make use of other MachineFunction fields, so I think it's best to
just categorically disallow dependency on the MachineFunction state in
the constructor and to always construct this at the same time as the
MachineFunction itself.

A missing feature I still could use is a way to access an custom
analysis pass on the IR here.
2022-12-21 10:49:32 -05:00

93 lines
3.2 KiB
C++

//===-- AVRMachineFuctionInfo.h - AVR 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 AVR-specific per-machine-function information.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_AVR_MACHINE_FUNCTION_INFO_H
#define LLVM_AVR_MACHINE_FUNCTION_INFO_H
#include "llvm/CodeGen/MachineFunction.h"
namespace llvm {
/// Contains AVR-specific information for each MachineFunction.
class AVRMachineFunctionInfo : public MachineFunctionInfo {
/// Indicates if a register has been spilled by the register
/// allocator.
bool HasSpills;
/// Indicates if there are any fixed size allocas present.
/// Note that if there are only variable sized allocas this is set to false.
bool HasAllocas;
/// Indicates if arguments passed using the stack are being
/// used inside the function.
bool HasStackArgs;
/// Whether or not the function is an interrupt handler.
bool IsInterruptHandler;
/// Whether or not the function is an non-blocking interrupt handler.
bool IsSignalHandler;
/// Size of the callee-saved register portion of the
/// stack frame in bytes.
unsigned CalleeSavedFrameSize;
/// FrameIndex for start of varargs area.
int VarArgsFrameIndex;
public:
AVRMachineFunctionInfo(const Function &F, const TargetSubtargetInfo *STI)
: HasSpills(false), HasAllocas(false), HasStackArgs(false),
CalleeSavedFrameSize(0), VarArgsFrameIndex(0) {
CallingConv::ID CallConv = F.getCallingConv();
this->IsInterruptHandler =
CallConv == CallingConv::AVR_INTR || F.hasFnAttribute("interrupt");
this->IsSignalHandler =
CallConv == CallingConv::AVR_SIGNAL || F.hasFnAttribute("signal");
}
MachineFunctionInfo *
clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF,
const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB)
const override {
return DestMF.cloneInfo<AVRMachineFunctionInfo>(*this);
}
bool getHasSpills() const { return HasSpills; }
void setHasSpills(bool B) { HasSpills = B; }
bool getHasAllocas() const { return HasAllocas; }
void setHasAllocas(bool B) { HasAllocas = B; }
bool getHasStackArgs() const { return HasStackArgs; }
void setHasStackArgs(bool B) { HasStackArgs = B; }
/// Checks if the function is some form of interrupt service routine.
bool isInterruptOrSignalHandler() const {
return isInterruptHandler() || isSignalHandler();
}
bool isInterruptHandler() const { return IsInterruptHandler; }
bool isSignalHandler() const { return IsSignalHandler; }
unsigned getCalleeSavedFrameSize() const { return CalleeSavedFrameSize; }
void setCalleeSavedFrameSize(unsigned Bytes) { CalleeSavedFrameSize = Bytes; }
int getVarArgsFrameIndex() const { return VarArgsFrameIndex; }
void setVarArgsFrameIndex(int Idx) { VarArgsFrameIndex = Idx; }
};
} // namespace llvm
#endif // LLVM_AVR_MACHINE_FUNCTION_INFO_H