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

Elam Cohavi llvmlistbot at llvm.org
Fri Aug 22 09:45:05 PDT 2025


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

This change swaps `std::map` for its LLVM equivalent type.

>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] [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});
 



More information about the Mlir-commits mailing list