forked from OSchip/llvm-project
111 lines
3.8 KiB
C++
111 lines
3.8 KiB
C++
//===--- RDFLiveness.h ----------------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Recalculate the liveness information given a data flow graph.
|
|
// This includes block live-ins and kill flags.
|
|
|
|
#ifndef RDF_LIVENESS_H
|
|
#define RDF_LIVENESS_H
|
|
|
|
#include "RDFGraph.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include <map>
|
|
|
|
using namespace llvm;
|
|
|
|
namespace llvm {
|
|
class MachineBasicBlock;
|
|
class MachineFunction;
|
|
class MachineRegisterInfo;
|
|
class TargetRegisterInfo;
|
|
class MachineDominatorTree;
|
|
class MachineDominanceFrontier;
|
|
|
|
namespace rdf {
|
|
struct Liveness {
|
|
public:
|
|
typedef std::map<MachineBasicBlock*,RegisterSet> LiveMapType;
|
|
typedef std::map<RegisterRef,NodeSet> RefMap;
|
|
|
|
Liveness(MachineRegisterInfo &mri, const DataFlowGraph &g)
|
|
: DFG(g), TRI(g.getTRI()), MDT(g.getDT()), MDF(g.getDF()),
|
|
RAI(g.getRAI()), MRI(mri), Empty(), Trace(false) {}
|
|
|
|
NodeList getAllReachingDefs(RegisterRef RefRR, NodeAddr<RefNode*> RefA,
|
|
bool FullChain = false, const RegisterSet &DefRRs = RegisterSet());
|
|
NodeList getAllReachingDefs(NodeAddr<RefNode*> RefA);
|
|
NodeSet getAllReachingDefsRec(RegisterRef RefRR, NodeAddr<RefNode*> RefA,
|
|
NodeSet &Visited, const NodeSet &Defs);
|
|
NodeSet getAllReachedUses(RegisterRef RefRR, NodeAddr<DefNode*> DefA,
|
|
const RegisterSet &DefRRs = RegisterSet());
|
|
|
|
LiveMapType &getLiveMap() { return LiveMap; }
|
|
const LiveMapType &getLiveMap() const { return LiveMap; }
|
|
const RefMap &getRealUses(NodeId P) const {
|
|
auto F = RealUseMap.find(P);
|
|
return F == RealUseMap.end() ? Empty : F->second;
|
|
}
|
|
|
|
void computePhiInfo();
|
|
void computeLiveIns();
|
|
void resetLiveIns();
|
|
void resetKills();
|
|
void resetKills(MachineBasicBlock *B);
|
|
|
|
void trace(bool T) { Trace = T; }
|
|
|
|
private:
|
|
const DataFlowGraph &DFG;
|
|
const TargetRegisterInfo &TRI;
|
|
const MachineDominatorTree &MDT;
|
|
const MachineDominanceFrontier &MDF;
|
|
const RegisterAliasInfo &RAI;
|
|
MachineRegisterInfo &MRI;
|
|
LiveMapType LiveMap;
|
|
const RefMap Empty;
|
|
bool Trace;
|
|
|
|
// Cache of mapping from node ids (for RefNodes) to the containing
|
|
// basic blocks. Not computing it each time for each node reduces
|
|
// the liveness calculation time by a large fraction.
|
|
typedef DenseMap<NodeId,MachineBasicBlock*> NodeBlockMap;
|
|
NodeBlockMap NBMap;
|
|
|
|
// Phi information:
|
|
//
|
|
// map: NodeId -> (map: RegisterRef -> NodeSet)
|
|
// phi id -> (map: register -> set of reached non-phi uses)
|
|
std::map<NodeId, RefMap> RealUseMap;
|
|
|
|
// Inverse iterated dominance frontier.
|
|
std::map<MachineBasicBlock*,std::set<MachineBasicBlock*>> IIDF;
|
|
|
|
// Live on entry.
|
|
std::map<MachineBasicBlock*,RefMap> PhiLON;
|
|
|
|
// Phi uses are considered to be located at the end of the block that
|
|
// they are associated with. The reaching def of a phi use dominates the
|
|
// block that the use corresponds to, but not the block that contains
|
|
// the phi itself. To include these uses in the liveness propagation (up
|
|
// the dominator tree), create a map: block -> set of uses live on exit.
|
|
std::map<MachineBasicBlock*,RefMap> PhiLOX;
|
|
|
|
bool isRestrictedToRef(NodeAddr<InstrNode*> IA, NodeAddr<RefNode*> RA,
|
|
RegisterRef RR) const;
|
|
RegisterRef getRestrictedRegRef(NodeAddr<RefNode*> RA) const;
|
|
unsigned getPhysReg(RegisterRef RR) const;
|
|
MachineBasicBlock *getBlockWithRef(NodeId RN) const;
|
|
void traverse(MachineBasicBlock *B, RefMap &LiveIn);
|
|
void emptify(RefMap &M);
|
|
};
|
|
} // namespace rdf
|
|
} // namespace llvm
|
|
|
|
#endif // RDF_LIVENESS_H
|