[Mlir-commits] [mlir] 2077fb8 - [mlir] Avoid repeated map lookups (NFC) (#113074)

[llvmlistbot at llvm.org]

Author: Kazu Hirata
Date: 2024-10-20T10:42:28-07:00
New Revision: 2077fb80ffb58cd1060ec6a5475399c6ad297df3

URL: https://github.com/llvm/llvm-project/commit/2077fb80ffb58cd1060ec6a5475399c6ad297df3
DIFF: https://github.com/llvm/llvm-project/commit/2077fb80ffb58cd1060ec6a5475399c6ad297df3.diff

LOG: [mlir] Avoid repeated map lookups (NFC) (#113074)

Added: 
    

Modified: 
    mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp b/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
index 7b1b1f383e6343..32e1dcbc2cce28 100644
--- a/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
+++ b/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
@@ -1160,9 +1160,8 @@ bool mlir::sparse_tensor::isBlockSparsity(AffineMap dimToLvl) {
     } else if (auto dimOp = dyn_cast<AffineDimExpr>(result)) {
       auto pos = dimOp.getPosition();
       // Expect dim to be unset.
-      if (coeffientMap.find(pos) != coeffientMap.end())
+      if (!coeffientMap.try_emplace(pos, 0).second)
         return false;
-      coeffientMap[pos] = 0;
     } else {
       return false;
     }