shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[SCEVPatternMatch] Add signed cst match; use in LV (NFC) (#154568)

Browse Source 
Add a m_scev_SpecificSInt for matching a sign-extended value, and use it
to improve some code in LoopVectorize.
This commit is contained in:
Ramkumar Ramachandra 2025-08-21 16:46:53 +01:00 committed by GitHub
parent 5c36fb3303
commit a96b78cf41
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
llvm/include/llvm/Analysis/ScalarEvolutionPatternMatch.h
View File

@ -116,6 +116,18 @@ struct is_specific_cst {
/// Match an SCEV constant with a plain unsigned integer. /// Match an SCEV constant with a plain unsigned integer.
inline cst_pred_ty<is_specific_cst> m_scev_SpecificInt(uint64_t V) { return V; } inline cst_pred_ty<is_specific_cst> m_scev_SpecificInt(uint64_t V) { return V; }
struct is_specific_signed_cst {
int64_t CV;
is_specific_signed_cst(int64_t C) : CV(C) {}
bool isValue(const APInt &C) const { return C.trySExtValue() == CV; }
};
/// Match an SCEV constant with a plain signed integer (sign-extended value will
/// be matched)
inline cst_pred_ty<is_specific_signed_cst> m_scev_SpecificSInt(int64_t V) {
return V;
}
struct bind_cst_ty { struct bind_cst_ty {
const APInt *&CR; const APInt *&CR;

18
llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
View File

@ -5918,23 +5918,13 @@ void LoopVectorizationCostModel::setVectorizedCallDecision(ElementCount VF) {
// TODO: do we need to figure out the cost of an extract to get the // TODO: do we need to figure out the cost of an extract to get the
// first lane? Or do we hope that it will be folded away? // first lane? Or do we hope that it will be folded away?
ScalarEvolution *SE = PSE.getSE(); ScalarEvolution *SE = PSE.getSE();
const auto *SAR = if (!match(SE->getSCEV(ScalarParam),
dyn_cast<SCEVAddRecExpr>(SE->getSCEV(ScalarParam)); m_scev_AffineAddRec(
m_SCEV(), m_scev_SpecificSInt(Param.LinearStepOrPos),
if (!SAR || SAR->getLoop() != TheLoop) { m_SpecificLoop(TheLoop))))
ParamsOk = false; ParamsOk = false;
break; break;
} }
const SCEVConstant *Step =
dyn_cast<SCEVConstant>(SAR->getStepRecurrence(*SE));
if (!Step ||
Step->getAPInt().getSExtValue() != Param.LinearStepOrPos)
ParamsOk = false;
break;
}
case VFParamKind::GlobalPredicate: case VFParamKind::GlobalPredicate:
break; break;
default: default: