[Mlir-commits] [mlir] a2661a9 - [mlir] Avoid repeated hash lookups (NFC) (#132332)

[llvmlistbot at llvm.org]

Author: Kazu Hirata
Date: 2025-03-21T08:01:51-07:00
New Revision: a2661a9904f3633e438bcdc3922d99ef72ef27fb

URL: https://github.com/llvm/llvm-project/commit/a2661a9904f3633e438bcdc3922d99ef72ef27fb
DIFF: https://github.com/llvm/llvm-project/commit/a2661a9904f3633e438bcdc3922d99ef72ef27fb.diff

LOG: [mlir] Avoid repeated hash lookups (NFC) (#132332)

Added: 
    

Modified: 
    mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp b/mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp
index 54bcc48ad8812..41cd7395b1836 100644
--- a/mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp
+++ b/mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp
@@ -612,11 +612,8 @@ bool mlir::affine::getFusionComputeCost(AffineForOp srcForOp,
         // 'insertPointParent'.
         getAffineForIVs(*user, &loops);
         if (llvm::is_contained(loops, cast<AffineForOp>(insertPointParent))) {
-          if (auto forOp = dyn_cast_or_null<AffineForOp>(user->getParentOp())) {
-            if (computeCostMap.count(forOp) == 0)
-              computeCostMap[forOp] = 0;
-            computeCostMap[forOp] -= 1;
-          }
+          if (auto forOp = dyn_cast_or_null<AffineForOp>(user->getParentOp()))
+            --computeCostMap[forOp];
         }
       }
     }