prune on boundaries

EnzymeAD · Nov 13, 2024 · a8dede8 · a8dede8
1 parent 5b94be0
commit a8dede8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/enzyme/Enzyme/Herbie.cpp b/enzyme/Enzyme/Herbie.cpp
@@ -3801,7 +3801,7 @@ bool accuracyDPSolver(
         if (currCompCost - otherCompCost >
                 std::fabs(FPOptCostDominanceThreshold *
                           otherCompCost.getValue().getValue()) &&
-            currAccCost - otherAccCost >
+            currAccCost - otherAccCost >=
                 std::fabs(FPOptAccuracyDominanceThreshold * otherAccCost)) {
           // if (EnzymePrintFPOpt)
           //   llvm::errs() << "AO candidate with computation cost: "
@@ -3901,7 +3901,7 @@ bool accuracyDPSolver(
         if (currCompCost - otherCompCost >
                 std::fabs(FPOptCostDominanceThreshold *
                           otherCompCost.getValue().getValue()) &&
-            currAccCost - otherAccCost >
+            currAccCost - otherAccCost >=
                 std::fabs(FPOptAccuracyDominanceThreshold * otherAccCost)) {
           // if (EnzymePrintFPOpt)
           //   llvm::errs() << "ACC candidate with computation cost: "