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

kazutakahirata · web-flow · commit a2661a9904f3 · 2025-03-21T08:01:51.000-07:00
diff --git 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];
         }
       }
     }

Original file line number	Diff line number	Diff line change
`@@ -612,11 +612,8 @@ bool mlir::affine::getFusionComputeCost(AffineForOp srcForOp,`
`612`	`612`	`// 'insertPointParent'.`
`613`	`613`	`getAffineForIVs(*user, &loops);`
`614`	`614`	`if (llvm::is_contained(loops, cast<AffineForOp>(insertPointParent))) {`
`615`		`- if (auto forOp = dyn_cast_or_null<AffineForOp>(user->getParentOp())) {`
`616`		`- if (computeCostMap.count(forOp) == 0)`
`617`		`- computeCostMap[forOp] = 0;`
`618`		`- computeCostMap[forOp] -= 1;`
`619`		`- }`
	`615`	`+ if (auto forOp = dyn_cast_or_null<AffineForOp>(user->getParentOp()))`
	`616`	`+ --computeCostMap[forOp];`
`620`	`617`	`}`
`621`	`618`	`}`
`622`	`619`	`}`