mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-12-25 21:16:19 +00:00
c0bb0e899d
OK, not really. We don't want to reintroduce the old rewriter hacks. This patch extracts virtual register rewriting as a separate pass that runs after the register allocator. This is possible now that CodeGen/Passes.cpp can configure the full optimizing register allocator pipeline. The rewriter pass uses register assignments in VirtRegMap to rewrite virtual registers to physical registers, and it inserts kill flags based on live intervals. These finalization steps are the same for the optimizing register allocators: RABasic, RAGreedy, and PBQP. llvm-svn: 158244
191 lines
6.6 KiB
C++
191 lines
6.6 KiB
C++
//===-- llvm/CodeGen/VirtRegMap.h - Virtual Register Map -*- C++ -*--------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements a virtual register map. This maps virtual registers to
|
|
// physical registers and virtual registers to stack slots. It is created and
|
|
// updated by a register allocator and then used by a machine code rewriter that
|
|
// adds spill code and rewrites virtual into physical register references.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CODEGEN_VIRTREGMAP_H
|
|
#define LLVM_CODEGEN_VIRTREGMAP_H
|
|
|
|
#include "llvm/CodeGen/MachineFunctionPass.h"
|
|
#include "llvm/Target/TargetRegisterInfo.h"
|
|
#include "llvm/ADT/IndexedMap.h"
|
|
|
|
namespace llvm {
|
|
class MachineInstr;
|
|
class MachineFunction;
|
|
class MachineRegisterInfo;
|
|
class TargetInstrInfo;
|
|
class raw_ostream;
|
|
class SlotIndexes;
|
|
|
|
class VirtRegMap : public MachineFunctionPass {
|
|
public:
|
|
enum {
|
|
NO_PHYS_REG = 0,
|
|
NO_STACK_SLOT = (1L << 30)-1,
|
|
MAX_STACK_SLOT = (1L << 18)-1
|
|
};
|
|
|
|
private:
|
|
MachineRegisterInfo *MRI;
|
|
const TargetInstrInfo *TII;
|
|
const TargetRegisterInfo *TRI;
|
|
MachineFunction *MF;
|
|
|
|
/// Virt2PhysMap - This is a virtual to physical register
|
|
/// mapping. Each virtual register is required to have an entry in
|
|
/// it; even spilled virtual registers (the register mapped to a
|
|
/// spilled register is the temporary used to load it from the
|
|
/// stack).
|
|
IndexedMap<unsigned, VirtReg2IndexFunctor> Virt2PhysMap;
|
|
|
|
/// Virt2StackSlotMap - This is virtual register to stack slot
|
|
/// mapping. Each spilled virtual register has an entry in it
|
|
/// which corresponds to the stack slot this register is spilled
|
|
/// at.
|
|
IndexedMap<int, VirtReg2IndexFunctor> Virt2StackSlotMap;
|
|
|
|
/// Virt2SplitMap - This is virtual register to splitted virtual register
|
|
/// mapping.
|
|
IndexedMap<unsigned, VirtReg2IndexFunctor> Virt2SplitMap;
|
|
|
|
/// createSpillSlot - Allocate a spill slot for RC from MFI.
|
|
unsigned createSpillSlot(const TargetRegisterClass *RC);
|
|
|
|
VirtRegMap(const VirtRegMap&); // DO NOT IMPLEMENT
|
|
void operator=(const VirtRegMap&); // DO NOT IMPLEMENT
|
|
|
|
public:
|
|
static char ID;
|
|
VirtRegMap() : MachineFunctionPass(ID), Virt2PhysMap(NO_PHYS_REG),
|
|
Virt2StackSlotMap(NO_STACK_SLOT), Virt2SplitMap(0) { }
|
|
virtual bool runOnMachineFunction(MachineFunction &MF);
|
|
|
|
virtual void getAnalysisUsage(AnalysisUsage &AU) const {
|
|
AU.setPreservesAll();
|
|
MachineFunctionPass::getAnalysisUsage(AU);
|
|
}
|
|
|
|
MachineFunction &getMachineFunction() const {
|
|
assert(MF && "getMachineFunction called before runOnMachineFunction");
|
|
return *MF;
|
|
}
|
|
|
|
MachineRegisterInfo &getRegInfo() const { return *MRI; }
|
|
const TargetRegisterInfo &getTargetRegInfo() const { return *TRI; }
|
|
|
|
void grow();
|
|
|
|
/// @brief returns true if the specified virtual register is
|
|
/// mapped to a physical register
|
|
bool hasPhys(unsigned virtReg) const {
|
|
return getPhys(virtReg) != NO_PHYS_REG;
|
|
}
|
|
|
|
/// @brief returns the physical register mapped to the specified
|
|
/// virtual register
|
|
unsigned getPhys(unsigned virtReg) const {
|
|
assert(TargetRegisterInfo::isVirtualRegister(virtReg));
|
|
return Virt2PhysMap[virtReg];
|
|
}
|
|
|
|
/// @brief creates a mapping for the specified virtual register to
|
|
/// the specified physical register
|
|
void assignVirt2Phys(unsigned virtReg, unsigned physReg) {
|
|
assert(TargetRegisterInfo::isVirtualRegister(virtReg) &&
|
|
TargetRegisterInfo::isPhysicalRegister(physReg));
|
|
assert(Virt2PhysMap[virtReg] == NO_PHYS_REG &&
|
|
"attempt to assign physical register to already mapped "
|
|
"virtual register");
|
|
Virt2PhysMap[virtReg] = physReg;
|
|
}
|
|
|
|
/// @brief clears the specified virtual register's, physical
|
|
/// register mapping
|
|
void clearVirt(unsigned virtReg) {
|
|
assert(TargetRegisterInfo::isVirtualRegister(virtReg));
|
|
assert(Virt2PhysMap[virtReg] != NO_PHYS_REG &&
|
|
"attempt to clear a not assigned virtual register");
|
|
Virt2PhysMap[virtReg] = NO_PHYS_REG;
|
|
}
|
|
|
|
/// @brief clears all virtual to physical register mappings
|
|
void clearAllVirt() {
|
|
Virt2PhysMap.clear();
|
|
grow();
|
|
}
|
|
|
|
/// @brief returns the register allocation preference.
|
|
unsigned getRegAllocPref(unsigned virtReg);
|
|
|
|
/// @brief returns true if VirtReg is assigned to its preferred physreg.
|
|
bool hasPreferredPhys(unsigned VirtReg) {
|
|
return getPhys(VirtReg) == getRegAllocPref(VirtReg);
|
|
}
|
|
|
|
/// @brief records virtReg is a split live interval from SReg.
|
|
void setIsSplitFromReg(unsigned virtReg, unsigned SReg) {
|
|
Virt2SplitMap[virtReg] = SReg;
|
|
}
|
|
|
|
/// @brief returns the live interval virtReg is split from.
|
|
unsigned getPreSplitReg(unsigned virtReg) const {
|
|
return Virt2SplitMap[virtReg];
|
|
}
|
|
|
|
/// getOriginal - Return the original virtual register that VirtReg descends
|
|
/// from through splitting.
|
|
/// A register that was not created by splitting is its own original.
|
|
/// This operation is idempotent.
|
|
unsigned getOriginal(unsigned VirtReg) const {
|
|
unsigned Orig = getPreSplitReg(VirtReg);
|
|
return Orig ? Orig : VirtReg;
|
|
}
|
|
|
|
/// @brief returns true if the specified virtual register is not
|
|
/// mapped to a stack slot or rematerialized.
|
|
bool isAssignedReg(unsigned virtReg) const {
|
|
if (getStackSlot(virtReg) == NO_STACK_SLOT)
|
|
return true;
|
|
// Split register can be assigned a physical register as well as a
|
|
// stack slot or remat id.
|
|
return (Virt2SplitMap[virtReg] && Virt2PhysMap[virtReg] != NO_PHYS_REG);
|
|
}
|
|
|
|
/// @brief returns the stack slot mapped to the specified virtual
|
|
/// register
|
|
int getStackSlot(unsigned virtReg) const {
|
|
assert(TargetRegisterInfo::isVirtualRegister(virtReg));
|
|
return Virt2StackSlotMap[virtReg];
|
|
}
|
|
|
|
/// @brief create a mapping for the specifed virtual register to
|
|
/// the next available stack slot
|
|
int assignVirt2StackSlot(unsigned virtReg);
|
|
/// @brief create a mapping for the specified virtual register to
|
|
/// the specified stack slot
|
|
void assignVirt2StackSlot(unsigned virtReg, int frameIndex);
|
|
|
|
void print(raw_ostream &OS, const Module* M = 0) const;
|
|
void dump() const;
|
|
};
|
|
|
|
inline raw_ostream &operator<<(raw_ostream &OS, const VirtRegMap &VRM) {
|
|
VRM.print(OS);
|
|
return OS;
|
|
}
|
|
} // End llvm namespace
|
|
|
|
#endif
|