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

[Kazu Hirata]

https://github.com/kazutakahirata created https://github.com/llvm/llvm-project/pull/132332

None

>From 1bd4889f99d633f3bc3f59df7cc7eae7ba799536 Mon Sep 17 00:00:00 2001
From: Kazu Hirata <kazu at google.com>
Date: Thu, 20 Mar 2025 09:46:44 -0700
Subject: [PATCH] [mlir] Avoid repeated hash lookups (NFC)

---
 mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

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];
         }
       }
     }