!fixup getResultType() in VPMulAccumulateReductionRecipe.

Author: ElvisWang123

llvm/lib/Transforms/Vectorize/VPlan.h

@@ -2686,7 +2686,7 @@ class VPMulAccumulateReductionRecipe : public VPReductionRecipe {
   bool IsNonNeg = false;
 
   /// The scalar type after extending.
-  Type *ResultTy;
+  Type *ResultTy = nullptr;
 
   /// For cloning VPMulAccumulateReductionRecipe.
   VPMulAccumulateReductionRecipe(VPMulAccumulateReductionRecipe *MulAcc)
@@ -2762,11 +2762,7 @@ class VPMulAccumulateReductionRecipe : public VPReductionRecipe {
              VPSlotTracker &SlotTracker) const override;
 #endif
 
-  Type *getResultType() const {
-    assert(isExtended() && "Only support getResultType when this recipe "
-                           "is implicitly extend.");
-    return ResultTy;
-  }
+  Type *getResultType() const { return ResultTy; }
 
   /// The first vector value to be extended and reduced.
   VPValue *getVecOp0() const { return getOperand(1); }

llvm/lib/Transforms/Vectorize/VPlanAnalysis.cpp

@@ -273,8 +273,12 @@ Type *VPTypeAnalysis::inferScalarType(const VPValue *V) {
             // TODO: Use info from interleave group.
             return V->getUnderlyingValue()->getType();
           })
-          .Case<VPExtendedReductionRecipe, VPMulAccumulateReductionRecipe>(
+          .Case<VPExtendedReductionRecipe>(
               [](const auto *R) { return R->getResultType(); })
+          .Case<VPMulAccumulateReductionRecipe>([this](const auto *R) {
+            return R->isExtended() ? R->getResultType()
+                                   : inferScalarType(R->getOperand(0));
+          })
           .Case<VPExpandSCEVRecipe>([](const VPExpandSCEVRecipe *R) {
             return R->getSCEV()->getType();
           })

llvm/lib/Transforms/Vectorize/VPlanTransforms.cpp

@@ -2530,8 +2530,10 @@ void VPlanTransforms::convertToConcreteRecipes(VPlan &Plan,
       ToRemove.push_back(VPI);
     }
     for (VPRecipeBase &R : make_early_inc_range(*VPBB)) {
-      if (auto *ExtRed = dyn_cast<VPExtendedReductionRecipe>(&R))
+      if (auto *ExtRed = dyn_cast<VPExtendedReductionRecipe>(&R)) {
         expandVPExtendedReduction(ExtRed);
+        continue;
+      }
       if (auto *MulAcc = dyn_cast<VPMulAccumulateReductionRecipe>(&R))
         expandVPMulAccumulateReduction(MulAcc);
     }