[Affine] Avoid repeated hash lookups (NFC) (#111226)

kazutakahirata · web-flow · commit fe9f1a215e0b · 2024-10-05T09:50:14.000-07:00
diff --git a/mlir/lib/Dialect/Affine/Transforms/PipelineDataTransfer.cpp b/mlir/lib/Dialect/Affine/Transforms/PipelineDataTransfer.cpp
@@ -350,8 +350,7 @@ void PipelineDataTransfer::runOnAffineForOp(AffineForOp forOp) {
   }
   // Everything else (including compute ops and dma finish) are shifted by one.
   for (auto &op : forOp.getBody()->without_terminator())
-    if (!instShiftMap.contains(&op))
-      instShiftMap[&op] = 1;
+    instShiftMap.try_emplace(&op, 1);
 
   // Get shifts stored in map.
   SmallVector<uint64_t, 8> shifts(forOp.getBody()->getOperations().size());

Original file line number	Diff line number	Diff line change
`@@ -350,8 +350,7 @@ void PipelineDataTransfer::runOnAffineForOp(AffineForOp forOp) {`
`350`	`350`	`}`
`351`	`351`	`// Everything else (including compute ops and dma finish) are shifted by one.`
`352`	`352`	`for (auto &op : forOp.getBody()->without_terminator())`
`353`		`- if (!instShiftMap.contains(&op))`
`354`		`- instShiftMap[&op] = 1;`
	`353`	`+ instShiftMap.try_emplace(&op, 1);`
`355`	`354`
`356`	`355`	`// Get shifts stored in map.`
`357`	`356`	`SmallVector<uint64_t, 8> shifts(forOp.getBody()->getOperations().size());`