[Mlir-commits] [mlir] [mlir][linalg][nfc] use `llvm::DenseMap` (PR #154981)

[Elam Cohavi]

https://github.com/elamc-2 updated https://github.com/llvm/llvm-project/pull/154981

>From 618cba4d368a0a3670128bae1ec6a7a93227673d Mon Sep 17 00:00:00 2001
From: Elam Cohavi <elam.cohavi at gmail.com>
Date: Fri, 22 Aug 2025 17:40:12 +0100
Subject: [PATCH 1/2] [mlir][linalg][nfc] use `llvm::DenseMap`

---
 .../Transforms/DecomposeGenericByUnfoldingPermutation.cpp      | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp b/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp
index b7da20c108f3e..9b5c14887b253 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp
@@ -8,7 +8,6 @@
 //
 #include "mlir/Dialect/Linalg/IR/Linalg.h"
 #include "mlir/Dialect/Linalg/Transforms/Transforms.h"
-#include <map>
 #include <utility>
 
 using namespace mlir;
@@ -125,7 +124,7 @@ computeTransposeBroadcast(AffineMap &map) {
     // refering to d3, d4. Therefore, re-base the transpose dimensions so
     // that they start from d0.
     permutation.resize(minorSize);
-    std::map<int64_t, int64_t> minorMap;
+    llvm::DenseMap<int64_t, int64_t> minorMap;
     for (int64_t i = 0; i < minorSize; ++i)
       minorMap.insert({sortedResMap[i], i});
 

>From 54944446c08b18c0a45ca5eac3552a91adbb0e44 Mon Sep 17 00:00:00 2001
From: Elam Cohavi <elam.cohavi at gmail.com>
Date: Mon, 25 Aug 2025 08:04:33 +0100
Subject: [PATCH 2/2] [mlir][linalg][nfc] map as `llvm::SmallVector`

---
 .../Transforms/DecomposeGenericByUnfoldingPermutation.cpp    | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp b/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp
index 9b5c14887b253..1c31b740c9919 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/DecomposeGenericByUnfoldingPermutation.cpp
@@ -124,9 +124,10 @@ computeTransposeBroadcast(AffineMap &map) {
     // refering to d3, d4. Therefore, re-base the transpose dimensions so
     // that they start from d0.
     permutation.resize(minorSize);
-    llvm::DenseMap<int64_t, int64_t> minorMap;
+    SmallVector<int64_t> minorMap;
+    minorMap.resize(minorSize);
     for (int64_t i = 0; i < minorSize; ++i)
-      minorMap.insert({sortedResMap[i], i});
+      minorMap[sortedResMap[i]] = i;
 
     // Re-map the dimensions.
     SmallVector<int64_t> remappedResult(minorSize);