Clear non-fp FMFs in adjustRecipesForReductions

Author: lukel97

llvm/lib/Transforms/Vectorize/LoopVectorize.cpp

@@ -9803,9 +9803,13 @@ void LoopVectorizationPlanner::adjustRecipesForReductions(
       if (CM.blockNeedsPredicationForAnyReason(BB))
         CondOp = RecipeBuilder.getBlockInMask(BB);
 
+      // Non-FP RdxDescs will have all fast math flags set, so clear them.
+      FastMathFlags FMFs = isa<FPMathOperator>(CurrentLinkI)
+                               ? RdxDesc.getFastMathFlags()
+                               : FastMathFlags();
       auto *RedRecipe = new VPReductionRecipe(
-          RdxDesc.getRecurrenceKind(), RdxDesc.getFastMathFlags(), CurrentLinkI,
-          PreviousLink, VecOp, CondOp, CM.useOrderedReductions(RdxDesc),
+          RdxDesc.getRecurrenceKind(), FMFs, CurrentLinkI, PreviousLink, VecOp,
+          CondOp, CM.useOrderedReductions(RdxDesc),
           CurrentLinkI->getDebugLoc());
       // Append the recipe to the end of the VPBasicBlock because we need to
       // ensure that it comes after all of it's inputs, including CondOp.

llvm/lib/Transforms/Vectorize/VPlan.h

@@ -2250,10 +2250,8 @@ class VPReductionRecipe : public VPRecipeWithIRFlags {
                     FastMathFlags FMFs, Instruction *I,
                     ArrayRef<VPValue *> Operands, VPValue *CondOp,
                     bool IsOrdered, DebugLoc DL)
-      : VPRecipeWithIRFlags(
-            SC, Operands,
-            isa_and_nonnull<FPMathOperator>(I) ? FMFs : FastMathFlags(), DL),
-        RdxKind(RdxKind), IsOrdered(IsOrdered) {
+      : VPRecipeWithIRFlags(SC, Operands, FMFs, DL), RdxKind(RdxKind),
+        IsOrdered(IsOrdered) {
     if (CondOp) {
       IsConditional = true;
       addOperand(CondOp);