#include "config.h"
#include "AirTmpWidth.h"
#if ENABLE(B3_JIT)
#include "AirCode.h"
#include "AirInstInlines.h"
#include "AirTmpWidthInlines.h"
#include <wtf/ListDump.h>
namespace JSC { namespace B3 { namespace Air {
TmpWidth::TmpWidth()
{
}
TmpWidth::TmpWidth(Code& code)
{
recompute<GP>(code);
recompute<FP>(code);
}
TmpWidth::~TmpWidth()
{
}
template <Bank bank>
void TmpWidth::recompute(Code& code)
{
constexpr bool beCareful = false;
constexpr bool verbose = false;
if (verbose) {
dataLogLn("Code before TmpWidth:");
dataLog(code);
}
auto& bankWidthsVector = widthsVector(bank);
bankWidthsVector.resize(AbsoluteTmpMapper<bank>::absoluteIndex(code.numTmps(bank)));
for (unsigned i = 0; i < bankWidthsVector.size(); ++i)
bankWidthsVector[i] = Widths(bank);
auto assumeTheWorst = [&] (Tmp tmp) {
if (bank == Arg(tmp).bank()) {
Width conservative = conservativeWidth(bank);
addWidths(tmp, { conservative, conservative });
}
};
RegisterSet::allRegisters().forEach(
[&] (Reg reg) {
assumeTheWorst(Tmp(reg));
});
if (beCareful) {
code.forAllTmps(assumeTheWorst);
}
Vector<Inst*> moves;
for (BasicBlock* block : code) {
for (Inst& inst : *block) {
if (inst.kind.opcode == Move && inst.args[1].isTmp()) {
if (Arg(inst.args[1]).bank() != bank)
continue;
if (inst.args[0].isTmp()) {
moves.append(&inst);
continue;
}
if (inst.args[0].isImm() && inst.args[0].value() >= 0) {
Tmp tmp = inst.args[1].tmp();
Widths& tmpWidths = widths(tmp);
Width maxWidth = Width64;
if (inst.args[0].value() <= std::numeric_limits<int8_t>::max())
maxWidth = Width8;
else if (inst.args[0].value() <= std::numeric_limits<int16_t>::max())
maxWidth = Width16;
else if (inst.args[0].value() <= std::numeric_limits<int32_t>::max())
maxWidth = Width32;
tmpWidths.def = std::max(tmpWidths.def, maxWidth);
continue;
}
}
inst.forEachTmp(
[&] (Tmp& tmp, Arg::Role role, Bank tmpBank, Width width) {
if (Arg(tmp).bank() != bank)
return;
Widths& tmpWidths = widths(tmp);
if (Arg::isAnyUse(role))
tmpWidths.use = std::max(tmpWidths.use, width);
if (Arg::isZDef(role))
tmpWidths.def = std::max(tmpWidths.def, width);
else if (Arg::isAnyDef(role))
tmpWidths.def = conservativeWidth(tmpBank);
});
}
}
bool changed = true;
while (changed) {
changed = false;
for (Inst* move : moves) {
ASSERT(move->kind.opcode == Move);
ASSERT(move->args[0].isTmp());
ASSERT(move->args[1].isTmp());
Widths& srcWidths = widths(move->args[0].tmp());
Widths& dstWidths = widths(move->args[1].tmp());
if (dstWidths.def < srcWidths.def) {
dstWidths.def = srcWidths.def;
changed = true;
}
if (srcWidths.use < dstWidths.use) {
srcWidths.use = dstWidths.use;
changed = true;
}
}
}
if (verbose) {
dataLogLn("bank: ", bank, ", widthsVector: ");
for (unsigned i = 0; i < bankWidthsVector.size(); ++i)
dataLogLn("\t", i, " : ", bankWidthsVector[i]);
}
}
void TmpWidth::Widths::dump(PrintStream& out) const
{
out.print("{use = ", use, ", def = ", def, "}");
}
} } }
#endif // ENABLE(B3_JIT)