AirRegLiveness.cpp [plain text]
#include "config.h"
#include "AirRegLiveness.h"
#if ENABLE(B3_JIT)
#include "AirArgInlines.h"
#include "AirInstInlines.h"
namespace JSC { namespace B3 { namespace Air {
RegLiveness::RegLiveness(Code& code)
: m_liveAtHead(code.size())
, m_liveAtTail(code.size())
, m_actions(code.size())
{
for (BasicBlock* block : code) {
ActionsForBoundary& actionsForBoundary = m_actions[block];
actionsForBoundary.resize(block->size() + 1);
for (size_t instIndex = block->size(); instIndex--;) {
Inst& inst = block->at(instIndex);
inst.forEach<Reg>(
[&] (Reg& reg, Arg::Role role, Bank, Width) {
if (Arg::isEarlyUse(role))
actionsForBoundary[instIndex].use.add(reg);
if (Arg::isEarlyDef(role))
actionsForBoundary[instIndex].def.add(reg);
if (Arg::isLateUse(role))
actionsForBoundary[instIndex + 1].use.add(reg);
if (Arg::isLateDef(role))
actionsForBoundary[instIndex + 1].def.add(reg);
});
}
}
for (BasicBlock* block : code) {
RegisterSet& liveAtTail = m_liveAtTail[block];
block->last().forEach<Reg>(
[&] (Reg& reg, Arg::Role role, Bank, Width) {
if (Arg::isLateUse(role))
liveAtTail.add(reg);
});
}
BitVector dirtyBlocks;
for (size_t blockIndex = code.size(); blockIndex--;)
dirtyBlocks.set(blockIndex);
bool changed;
do {
changed = false;
for (size_t blockIndex = code.size(); blockIndex--;) {
BasicBlock* block = code[blockIndex];
if (!block)
continue;
if (!dirtyBlocks.quickClear(blockIndex))
continue;
LocalCalc localCalc(*this, block);
for (size_t instIndex = block->size(); instIndex--;)
localCalc.execute(instIndex);
block->at(0).forEach<Reg>(
[&] (Reg& reg, Arg::Role role, Bank, Width) {
if (Arg::isEarlyDef(role))
localCalc.m_workset.remove(reg);
});
RegisterSet& liveAtHead = m_liveAtHead[block];
if (liveAtHead.subsumes(localCalc.m_workset))
continue;
liveAtHead.merge(localCalc.m_workset);
for (BasicBlock* predecessor : block->predecessors()) {
RegisterSet& liveAtTail = m_liveAtTail[predecessor];
if (liveAtTail.subsumes(localCalc.m_workset))
continue;
liveAtTail.merge(localCalc.m_workset);
dirtyBlocks.quickSet(predecessor->index());
changed = true;
}
}
} while (changed);
}
RegLiveness::~RegLiveness()
{
}
RegLiveness::LocalCalcForUnifiedTmpLiveness::LocalCalcForUnifiedTmpLiveness(UnifiedTmpLiveness& liveness, BasicBlock* block)
: LocalCalcBase(block)
, m_code(liveness.code)
, m_actions(liveness.actions[block])
{
for (Tmp tmp : liveness.liveAtTail(block)) {
if (tmp.isReg())
m_workset.add(tmp.reg());
}
}
void RegLiveness::LocalCalcForUnifiedTmpLiveness::execute(unsigned instIndex)
{
for (unsigned index : m_actions[instIndex + 1].def) {
Tmp tmp = Tmp::tmpForLinearIndex(m_code, index);
if (tmp.isReg())
m_workset.remove(tmp.reg());
}
for (unsigned index : m_actions[instIndex].use) {
Tmp tmp = Tmp::tmpForLinearIndex(m_code, index);
if (tmp.isReg())
m_workset.add(tmp.reg());
}
}
} } }
#endif // ENABLE(B3_JIT)