Fixup! Remove legacy model query

Author: ElvisWang123

llvm/lib/Transforms/Vectorize/LoopVectorize.cpp

@@ -7137,14 +7137,6 @@ bool VPCostContext::skipCostComputation(Instruction *UI, bool IsVector) const {
          SkipCostComputation.contains(UI);
 }
 
-bool VPCostContext::isInLoopReduction(const Instruction *UI, ElementCount VF,
-                                      Type *VectorTy) const {
-  return CM
-      .getReductionPatternCost(const_cast<Instruction *>(UI), VF, VectorTy,
-                               TTI::TCK_RecipThroughput)
-      .has_value();
-}
-
 InstructionCost
 LoopVectorizationPlanner::precomputeCosts(VPlan &Plan, ElementCount VF,
                                           VPCostContext &CostCtx) const {

llvm/lib/Transforms/Vectorize/VPlan.h

@@ -718,10 +718,6 @@ struct VPCostContext {
   /// Return true if the cost for \p UI shouldn't be computed, e.g. because it
   /// has already been pre-computed.
   bool skipCostComputation(Instruction *UI, bool IsVector) const;
-
-  /// Return true if the \p UI is part of the in-loop reduction.
-  bool isInLoopReduction(const Instruction *UI, ElementCount VF,
-                         Type *VectorTy) const;
 };
 
 /// VPRecipeBase is a base class modeling a sequence of one or more output IR

llvm/lib/Transforms/Vectorize/VPlanRecipes.cpp

@@ -2079,13 +2079,14 @@ InstructionCost VPReductionRecipe::computeCost(ElementCount VF,
   TTI::TargetCostKind CostKind = TTI::TCK_RecipThroughput;
   unsigned Opcode = RdxDesc.getOpcode();
 
-  // TODO: Support any-of reduction and in-loop reductions.
+  // TODO: Support any-of and in-loop reductions.
   assert(
       (!RecurrenceDescriptor::isAnyOfRecurrenceKind(RdxKind) ||
        ForceTargetInstructionCost.getNumOccurrences() > 0) &&
       "Any-of reduction not implemented in VPlan-based cost model currently.");
+  auto *ReductionPHIRecipe = dyn_cast<VPReductionPHIRecipe>(getOperand(0));
   assert(
-      (!Ctx.isInLoopReduction(getUnderlyingInstr(), VF, VectorTy) ||
+      (ReductionPHIRecipe && !ReductionPHIRecipe->isInLoop() ||
        ForceTargetInstructionCost.getNumOccurrences() > 0) &&
       "In-loop reduction not implemented in VPlan-based cost model currently.");