Skip to content

Commit

Permalink
Generate inline superclass test for Class.isAssignableFrom on Z
Browse files Browse the repository at this point in the history 
adds superclass check
checks for class depth at compile time to skip over superclass test for
the case when toClassDepth > fromClassDepth

Signed-off-by: Matthew Hall <matthew.hall3@outlook.com>
  • Loading branch information
@matthewhall2
matthewhall2 committed Dec 19, 2024
1 parent 85b9c02 commit 32f0761
Showing 1 changed file with 88 additions and 20 deletions.
108 changes: 88 additions & 20 deletions runtime/compiler/z/codegen/J9TreeEvaluator.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11573,6 +11573,53 @@ J9::Z::TreeEvaluator::VMarrayCheckEvaluator(TR::Node *node, TR::CodeGenerator *c
return 0;
}


static bool isInterfaceOrAbstract(TR::Node *clazz, TR::Compilation *comp)
{
while (clazz->getOpCodeValue() == TR::aloadi && clazz->getNumChildren() > 0)
{
clazz = clazz->getFirstChild();
}

// we don't assume its an interface if its not loadaddr
// (could have Class.forName(<someinterface>) as the node)
if (clazz->getOpCodeValue() != TR::loadaddr) return false;

TR::SymbolReference *thisClassSymRef = clazz->getSymbolReference();
return thisClassSymRef->isClassInterface(comp) || thisClassSymRef->isClassAbstract(comp);
}

static int32_t getCompileTimeClassDepth(TR::Node *clazz, TR::Compilation *comp)
{
int32_t classDepth = -1;
while (clazz->getOpCodeValue() == TR::aloadi && clazz->getNumChildren() > 0 && clazz->getFirstChild()->getOpCodeValue() == TR::aloadi)
{
clazz = clazz->getFirstChild();
}

if (clazz->getOpCodeValue() != TR::aloadi || clazz->getSymbolReference() != comp->getSymRefTab()->findJavaLangClassFromClassSymbolRef() ||
clazz->getFirstChild()->getOpCodeValue() != TR::loadaddr) return classDepth;

TR::Node *castClassRef = clazz->getFirstChild();

TR::SymbolReference *castClassSymRef = NULL;
if(castClassRef->getOpCode().hasSymbolReference())
castClassSymRef= castClassRef->getSymbolReference();

TR::StaticSymbol *castClassSym = NULL;
if (castClassSymRef && !castClassSymRef->isUnresolved())
castClassSym= castClassSymRef ? castClassSymRef->getSymbol()->getStaticSymbol() : NULL;

TR_OpaqueClassBlock * clazzz = NULL;
if (castClassSym)
clazzz = (TR_OpaqueClassBlock *) castClassSym->getStaticAddress();

if(clazzz)
classDepth = (int32_t)TR::Compiler->cls.classDepthOf(clazzz);

return classDepth;
}

/////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////////
static bool inlineIsAssignableFrom(TR::Node *node, TR::CodeGenerator *cg)
Expand Down Expand Up @@ -11727,51 +11774,72 @@ static bool inlineIsAssignableFrom(TR::Node *node, TR::CodeGenerator *cg)

TR::Register *J9::Z::TreeEvaluator::inlineCheckAssignableFromEvaluator(TR::Node *node, TR::CodeGenerator *cg)
{
TR::Register *fromClassReg = cg->evaluate(node->getFirstChild());
TR::Register *toClassReg = cg->evaluate(node->getSecondChild());

TR::Register *resultReg = cg->allocateRegister();
TR::Node *fromClass = node->getFirstChild();
TR::Node *toClass = node->getSecondChild();
TR::Register *fromClassReg = cg->evaluate(fromClass);
TR::Register *toClassReg = cg->evaluate(toClass);


TR::LabelSymbol *helperCallLabel = generateLabelSymbol(cg);
TR::LabelSymbol *doneLabel = generateLabelSymbol(cg);
TR::LabelSymbol *failLabel = generateLabelSymbol(cg);
TR::LabelSymbol *successLabel = generateLabelSymbol(cg);

TR::LabelSymbol* cFlowRegionStart = generateLabelSymbol(cg);
TR_S390ScratchRegisterManager *srm = cg->generateScratchRegisterManager(2);
TR::Register *sr1 = srm->findOrCreateScratchRegister();
TR::Register *sr2 = srm->findOrCreateScratchRegister();

TR::RegisterDependencyConditions* deps = new (cg->trHeapMemory()) TR::RegisterDependencyConditions(0, 5, cg);
deps->addPostCondition(fromClassReg, TR::RealRegister::AssignAny);
deps->addPostConditionIfNotAlreadyInserted(toClassReg, TR::RealRegister::AssignAny);
srm->addScratchRegistersToDependencyList(deps);


generateS390LabelInstruction(cg, TR::InstOpCode::label, node, cFlowRegionStart);
cFlowRegionStart->setStartInternalControlFlow();

/*
* check for class equality
* if equal, we are done. If not, fall through to helper call
*/
generateS390CompareAndBranchInstruction(cg, TR::InstOpCode::getCmpRegOpCode(), node, toClassReg, fromClassReg, TR::InstOpCode::COND_BE, successLabel, false, false);

/*
* TODO: add inlined tests (SuperclassTest, cast class cache, etc)
* Inlined tests will be used when possible, or will jump to the OOL section
* and perform the tests using the CHelper when not possible
*/
if (!isInterfaceOrAbstract(toClass, cg->comp()))
{
auto toClassDepth = getCompileTimeClassDepth(toClass, cg->comp());
if (!isInterfaceOrAbstract(fromClass, cg->comp()))
{
auto fromClassDepth = getCompileTimeClassDepth(fromClass, cg->comp());
if (toClassDepth > -1 && fromClassDepth > -1 && toClassDepth > fromClassDepth)
{
generateS390BranchInstruction(cg, TR::InstOpCode::BRC, TR::InstOpCode::COND_BRC, node, failLabel);
}
}
genTestIsSuper(cg, node, fromClassReg, toClassReg, sr1, sr2, NULL, NULL, toClassDepth, failLabel, doneLabel, helperCallLabel, deps, NULL, false, NULL, NULL);
generateS390BranchInstruction(cg, TR::InstOpCode::BRC, TR::InstOpCode::COND_BE, node, successLabel);
}


generateS390BranchInstruction(cg, TR::InstOpCode::BRC, TR::InstOpCode::COND_BRC, node, helperCallLabel);
TR_S390OutOfLineCodeSection *outlinedSlowPath = new (cg->trHeapMemory()) TR_S390OutOfLineCodeSection(helperCallLabel, doneLabel, cg);
cg->getS390OutOfLineCodeSectionList().push_front(outlinedSlowPath);
outlinedSlowPath->swapInstructionListsWithCompilation();

generateS390LabelInstruction(cg, TR::InstOpCode::label, node, helperCallLabel);
resultReg = TR::TreeEvaluator::performCall(node, false, cg);
TR::Register *resultReg = TR::TreeEvaluator::performCall(node, false, cg);

generateS390BranchInstruction(cg, TR::InstOpCode::BRC, TR::InstOpCode::COND_BRC, node, doneLabel); // exit OOL section
outlinedSlowPath->swapInstructionListsWithCompilation();

generateS390LabelInstruction(cg, TR::InstOpCode::label, node, failLabel);
generateRIInstruction(cg, TR::InstOpCode::LHI, node, resultReg, 0);
generateS390BranchInstruction(cg, TR::InstOpCode::BRC, TR::InstOpCode::COND_BRC, node, doneLabel);

generateS390LabelInstruction(cg, TR::InstOpCode::label, node, successLabel);
generateRIInstruction(cg, TR::InstOpCode::getLoadHalfWordImmOpCode(), node, resultReg, 1);
generateRIInstruction(cg, TR::InstOpCode::LHI, node, resultReg, 1);

TR::RegisterDependencyConditions* deps = new (cg->trHeapMemory()) TR::RegisterDependencyConditions(0, 3, cg);
deps->addPostCondition(fromClassReg, TR::RealRegister::AssignAny);
deps->addPostConditionIfNotAlreadyInserted(toClassReg, TR::RealRegister::AssignAny);
deps->addPostCondition(resultReg, TR::RealRegister::AssignAny);

deps->addPostCondition(resultReg, TR::RealRegister::AssignAny);
generateS390LabelInstruction(cg, TR::InstOpCode::label, node, doneLabel, deps);
doneLabel->setEndInternalControlFlow();

srm->stopUsingRegisters();
node->setRegister(resultReg);
return resultReg;
}
Expand Down

0 comments on commit 32f0761

Please sign in to comment.