[Mlir-commits] [mlir] [mlir] Add inliner interface to the index dialect (PR #170459)

[llvmlistbot at llvm.org]

https://github.com/jack-slingsby updated https://github.com/llvm/llvm-project/pull/170459

>From cac52d791b06c8dd27fb72ce6cd10204bfc649de Mon Sep 17 00:00:00 2001
From: Jack Slingsby <jack.slingsby at imgtec.com>
Date: Tue, 2 Dec 2025 18:05:35 +0000
Subject: [PATCH 1/2] Allow index dialect to be inlined

---
 mlir/lib/Dialect/Index/IR/IndexDialect.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/mlir/lib/Dialect/Index/IR/IndexDialect.cpp b/mlir/lib/Dialect/Index/IR/IndexDialect.cpp
index 183d0e33b2523..58589cc3fc5bf 100644
--- a/mlir/lib/Dialect/Index/IR/IndexDialect.cpp
+++ b/mlir/lib/Dialect/Index/IR/IndexDialect.cpp
@@ -8,6 +8,7 @@
 
 #include "mlir/Dialect/Index/IR/IndexDialect.h"
 #include "mlir/Conversion/ConvertToLLVM/ToLLVMInterface.h"
+#include "mlir/Transforms/InliningUtils.h"
 
 using namespace mlir;
 using namespace mlir::index;
@@ -15,10 +16,23 @@ using namespace mlir::index;
 //===----------------------------------------------------------------------===//
 // IndexDialect
 //===----------------------------------------------------------------------===//
+namespace {
+/// This class defines the interface for handling inlining for index
+/// dialect operations.
+struct IndexInlinerInterface : public DialectInlinerInterface {
+  using DialectInlinerInterface::DialectInlinerInterface;
+
+  /// All index dialect ops can be inlined.
+  bool isLegalToInline(Operation *, Region *, bool, IRMapping &) const final {
+    return true;
+  }
+};
+}
 
 void IndexDialect::initialize() {
   registerAttributes();
   registerOperations();
+  addInterfaces<IndexInlinerInterface>();
   declarePromisedInterface<ConvertToLLVMPatternInterface, IndexDialect>();
 }
 

>From 4dbcdece65c4663879426d6c444b084f0584631e Mon Sep 17 00:00:00 2001
From: Jack Slingsby <jack.slingsby at imgtec.com>
Date: Thu, 4 Dec 2025 10:30:19 +0000
Subject: [PATCH 2/2] Fix formatting issues

---
 mlir/lib/Dialect/Index/IR/IndexDialect.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlir/lib/Dialect/Index/IR/IndexDialect.cpp b/mlir/lib/Dialect/Index/IR/IndexDialect.cpp
index 58589cc3fc5bf..887e8e1ae78b6 100644
--- a/mlir/lib/Dialect/Index/IR/IndexDialect.cpp
+++ b/mlir/lib/Dialect/Index/IR/IndexDialect.cpp
@@ -27,7 +27,7 @@ struct IndexInlinerInterface : public DialectInlinerInterface {
     return true;
   }
 };
-}
+} // namespace
 
 void IndexDialect::initialize() {
   registerAttributes();