[llvm] [VPlan] Impl VPlan-based pattern match for ExtendedRed and MulAccRed (NFCI) (PR #113903)
Elvis Wang via llvm-commits
llvm-commits at lists.llvm.org
Mon Dec 23 01:38:11 PST 2024
================
@@ -2694,6 +2772,201 @@ class VPReductionEVLRecipe : public VPReductionRecipe {
}
};
+/// A recipe to represent inloop extended reduction operations, performing a
+/// reduction on a extended vector operand into a scalar value, and adding the
+/// result to a chain. This recipe is abstract and needs to be lowered to
+/// concrete recipes before codegen. The Operands are {ChainOp, VecOp,
+/// [Condition]}.
+class VPExtendedReductionRecipe : public VPReductionRecipe {
+ Instruction::CastOps ExtOp;
+ DebugLoc RedDL;
+
+protected:
+ VPExtendedReductionRecipe(const unsigned char SC,
+ const RecurrenceDescriptor &R, Instruction *RedI,
+ Instruction::CastOps ExtOp, DebugLoc ExtDL,
+ bool IsNonNeg, VPValue *ChainOp, VPValue *VecOp,
+ VPValue *CondOp, bool IsOrdered)
+ : VPReductionRecipe(SC, R, ArrayRef<VPValue *>({ChainOp, VecOp}), CondOp,
+ IsOrdered, IsNonNeg, ExtDL),
+ ExtOp(ExtOp), RedDL(RedI->getDebugLoc()) {}
+
+public:
+ VPExtendedReductionRecipe(const RecurrenceDescriptor &R, Instruction *RedI,
+ VPValue *ChainOp, VPWidenCastRecipe *Ext,
+ VPValue *CondOp, bool IsOrdered)
+ : VPExtendedReductionRecipe(VPDef::VPExtendedReductionSC, R, RedI,
+ Ext->getOpcode(), Ext->getDebugLoc(),
+ Ext->isNonNeg(), ChainOp, Ext->getOperand(0),
+ CondOp, IsOrdered) {}
+
+ ~VPExtendedReductionRecipe() override = default;
+
+ VPExtendedReductionRecipe *clone() override {
+ llvm_unreachable("Not implement yet");
----------------
ElvisWang123 wrote:
Implemented, thanks.
https://github.com/llvm/llvm-project/pull/113903
More information about the llvm-commits
mailing list