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

[llvmlistbot at llvm.org]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlir-affine

Author: Kazu Hirata (kazutakahirata)

<details>
<summary>Changes</summary>



---
Full diff: https://github.com/llvm/llvm-project/pull/132332.diff


1 Files Affected:

- (modified) mlir/lib/Dialect/Affine/Utils/LoopFusionUtils.cpp (+2-5) 


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

``````````

</details>


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