intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-25 19:44:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RISCV][GISel] Attempt to simplify how we handle type legality for F and D extensions. (#72174)

Browse Source 
Add helper that creates a lambda similar to typeIs, but containing the
predicate check for hasStdExtF and hasStdD. We can use this with legalIf
and all to reduce the number of manual lambdas we need to write.
This commit is contained in:
Craig Topper
2023-11-13 19:19:56 -08:00
committed by GitHub
parent 725115d7bb
commit 3161102f10
1 changed files with 16 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
llvm/lib/Target/RISCV/GISel/RISCVLegalizerInfo.cpp
View File

@@ -23,6 +23,17 @@
using namespace llvm;
using namespace LegalityPredicates;
// Is this type supported by scalar FP arithmetic operations given the current
// subtarget.
static LegalityPredicate typeIsScalarFPArith(unsigned TypeIdx,
const RISCVSubtarget &ST) {
return [=, &ST](const LegalityQuery &Query) {
return Query.Types[TypeIdx].isScalar() &&
((ST.hasStdExtF() && Query.Types[TypeIdx].getSizeInBits() == 32) ||
(ST.hasStdExtD() && Query.Types[TypeIdx].getSizeInBits() == 64));
};
}
RISCVLegalizerInfo::RISCVLegalizerInfo(const RISCVSubtarget &ST) {
const unsigned XLen = ST.getXLen();
const LLT sXLen = LLT::scalar(XLen);
@@ -212,10 +223,7 @@ RISCVLegalizerInfo::RISCVLegalizerInfo(const RISCVSubtarget &ST) {
getActionDefinitionsBuilder({G_FADD, G_FSUB, G_FMUL, G_FDIV, G_FMA, G_FNEG,
G_FABS, G_FSQRT, G_FMAXNUM, G_FMINNUM})
.legalIf([=, &ST](const LegalityQuery &Query) -> bool {
return (ST.hasStdExtF() && typeIs(0, s32)(Query)) ||
(ST.hasStdExtD() && typeIs(0, s64)(Query));
});
.legalIf(typeIsScalarFPArith(0, ST));
getActionDefinitionsBuilder(G_FPTRUNC).legalIf(
[=, &ST](const LegalityQuery &Query) -> bool {
@@ -229,34 +237,18 @@ RISCVLegalizerInfo::RISCVLegalizerInfo(const RISCVSubtarget &ST) {
});
getActionDefinitionsBuilder(G_FCMP)
.legalIf([=, &ST](const LegalityQuery &Query) -> bool {
return typeIs(0, sXLen)(Query) &&
((ST.hasStdExtF() && typeIs(1, s32)(Query)) ||
(ST.hasStdExtD() && typeIs(1, s64)(Query)));
})
.legalIf(all(typeIs(0, sXLen), typeIsScalarFPArith(1, ST)))
.clampScalar(0, sXLen, sXLen);
getActionDefinitionsBuilder(G_FCONSTANT)
.legalIf([=, &ST](const LegalityQuery &Query) -> bool {
return (ST.hasStdExtF() && typeIs(0, s32)(Query)) ||
(ST.hasStdExtD() && typeIs(0, s64)(Query));
});
getActionDefinitionsBuilder(G_FCONSTANT).legalIf(typeIsScalarFPArith(0, ST));
getActionDefinitionsBuilder({G_FPTOSI, G_FPTOUI})
.legalIf([=, &ST](const LegalityQuery &Query) -> bool {
return typeInSet(0, {s32, sXLen})(Query) &&
((ST.hasStdExtF() && typeIs(1, s32)(Query)) ||
(ST.hasStdExtD() && typeIs(1, s64)(Query)));
})
.legalIf(all(typeInSet(0, {s32, sXLen}), typeIsScalarFPArith(1, ST)))
.widenScalarToNextPow2(0)
.clampScalar(0, s32, sXLen);
getActionDefinitionsBuilder({G_SITOFP, G_UITOFP})
.legalIf([=, &ST](const LegalityQuery &Query) -> bool {
return ((ST.hasStdExtF() && typeIs(0, s32)(Query)) ||
(ST.hasStdExtD() && typeIs(0, s64)(Query))) &&
typeInSet(1, {s32, sXLen})(Query);
})
.legalIf(all(typeIsScalarFPArith(0, ST), typeInSet(1, {s32, sXLen})))
.widenScalarToNextPow2(1)
.clampScalar(1, s32, sXLen);