[Mlir-commits] [mlir] 65a5b18 - [Shape] Avoid repeated hash lookups (NFC) (#108926)

[llvmlistbot at llvm.org]

Author: Kazu Hirata
Date: 2024-09-17T00:18:33-07:00
New Revision: 65a5b18aa04087ffba11760bd4654aaae211aa38

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

LOG: [Shape] Avoid repeated hash lookups (NFC) (#108926)

Added: 
    

Modified: 
    mlir/lib/Dialect/Shape/Transforms/OutlineShapeComputation.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Shape/Transforms/OutlineShapeComputation.cpp b/mlir/lib/Dialect/Shape/Transforms/OutlineShapeComputation.cpp
index f467123f9b33de..3297740a5f3ae6 100644
--- a/mlir/lib/Dialect/Shape/Transforms/OutlineShapeComputation.cpp
+++ b/mlir/lib/Dialect/Shape/Transforms/OutlineShapeComputation.cpp
@@ -292,10 +292,8 @@ void OutlineShapeComputationPass::getClusterFromValue(
       cluster.insert(op);
       for (Value inp : op->getOperands()) {
         Operation *inpDefOp = inp.getDefiningOp();
-        if (nullptr != inpDefOp && !visited.contains(inpDefOp)) {
-          visited.insert(inpDefOp);
+        if (nullptr != inpDefOp && visited.insert(inpDefOp).second)
           queue.push(inpDefOp);
-        }
       }
     }
   }