Skip to content

Commit 2077fb8

Browse files
[mlir] Avoid repeated map lookups (NFC) (#113074)
1 parent 3bddf85 commit 2077fb8

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,9 +1160,8 @@ bool mlir::sparse_tensor::isBlockSparsity(AffineMap dimToLvl) {
11601160
} else if (auto dimOp = dyn_cast<AffineDimExpr>(result)) {
11611161
auto pos = dimOp.getPosition();
11621162
// Expect dim to be unset.
1163-
if (coeffientMap.find(pos) != coeffientMap.end())
1163+
if (!coeffientMap.try_emplace(pos, 0).second)
11641164
return false;
1165-
coeffientMap[pos] = 0;
11661165
} else {
11671166
return false;
11681167
}

0 commit comments

Comments
 (0)