[Mlir-commits] [mlir] [MLIR][Python] add builtin module transform test (PR #176388)

[Maksim Levental]

https://github.com/makslevental updated https://github.com/llvm/llvm-project/pull/176388

>From 83ffe1edac235bcbd8253e27ceb360d09276b98d Mon Sep 17 00:00:00 2001
From: Maksim Levental <maksim.levental at gmail.com>
Date: Fri, 16 Jan 2026 06:12:08 -0800
Subject: [PATCH 1/4] [MLIR][Python] add builtin module transform test

---
 .../python/dialects/transform_interpreter.py  | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/mlir/test/python/dialects/transform_interpreter.py b/mlir/test/python/dialects/transform_interpreter.py
index ca9ce5dbd23c1..36ab3c29d45d2 100644
--- a/mlir/test/python/dialects/transform_interpreter.py
+++ b/mlir/test/python/dialects/transform_interpreter.py
@@ -159,3 +159,22 @@ def repeated_include():
         interp.copy_symbols_and_merge_into(main, callee2)
     except ValueError as e:
         assert "doubly defined symbol @callee2" in str(e)
+
+
+ at test_in_context
+def check_builtin():
+    module = builtin_d.ModuleOp()
+    with module.context, ir.Location.unknown():
+        transform_module = builtin_d.Module.create()
+        transform_module.operation.attributes["transform.with_named_sequence"] = (
+            ir.UnitAttr.get()
+        )
+        with ir.InsertionPoint(transform_module.body):
+            named_sequence = NamedSequenceOp("__transform_main", [any_op_t()], [])
+            with ir.InsertionPoint(named_sequence.body):
+                YieldOp([])
+        interp.apply_named_sequence(
+            module,
+            transform_module.body.operations[0],
+            transform_module,
+        )

>From df63a0291e83875929a18558208ed853be66bb8d Mon Sep 17 00:00:00 2001
From: Maksim Levental <maksim.levental at gmail.com>
Date: Fri, 16 Jan 2026 07:30:37 -0800
Subject: [PATCH 2/4] Update transform_interpreter.py

---
 mlir/test/python/dialects/transform_interpreter.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/mlir/test/python/dialects/transform_interpreter.py b/mlir/test/python/dialects/transform_interpreter.py
index 36ab3c29d45d2..13699e0d8b2e2 100644
--- a/mlir/test/python/dialects/transform_interpreter.py
+++ b/mlir/test/python/dialects/transform_interpreter.py
@@ -1,7 +1,13 @@
 # RUN: %PYTHON %s | FileCheck %s
 
 from mlir import ir
-from mlir.dialects.transform import interpreter as interp
+from mlir.dialects.transform import (
+    interpreter as interp,
+    NamedSequenceOp,
+    any_op_t,
+    YieldOp,
+)
+from mlir.dialects import builtin as builtin_d
 
 
 def test_in_context(f):
@@ -173,6 +179,7 @@ def check_builtin():
             named_sequence = NamedSequenceOp("__transform_main", [any_op_t()], [])
             with ir.InsertionPoint(named_sequence.body):
                 YieldOp([])
+        transform_module.operation.verify()
         interp.apply_named_sequence(
             module,
             transform_module.body.operations[0],

>From 5d7975fd603b31bc76ba7728638f41a57d9651ef Mon Sep 17 00:00:00 2001
From: Maksim Levental <maksim.levental at gmail.com>
Date: Fri, 16 Jan 2026 07:31:31 -0800
Subject: [PATCH 3/4] Update transform_interpreter.py

---
 mlir/test/python/dialects/transform_interpreter.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlir/test/python/dialects/transform_interpreter.py b/mlir/test/python/dialects/transform_interpreter.py
index 13699e0d8b2e2..0553ed4070157 100644
--- a/mlir/test/python/dialects/transform_interpreter.py
+++ b/mlir/test/python/dialects/transform_interpreter.py
@@ -185,3 +185,4 @@ def check_builtin():
             transform_module.body.operations[0],
             transform_module,
         )
+        module.operation.verify()

>From d66d89419564f52d3162c48bd45c08de51536134 Mon Sep 17 00:00:00 2001
From: Maksim Levental <maksim.levental at gmail.com>
Date: Fri, 16 Jan 2026 07:41:41 -0800
Subject: [PATCH 4/4] Update transform_interpreter.py

---
 mlir/test/python/dialects/transform_interpreter.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/mlir/test/python/dialects/transform_interpreter.py b/mlir/test/python/dialects/transform_interpreter.py
index 0553ed4070157..491afb3f8bd5a 100644
--- a/mlir/test/python/dialects/transform_interpreter.py
+++ b/mlir/test/python/dialects/transform_interpreter.py
@@ -172,9 +172,9 @@ def check_builtin():
     module = builtin_d.ModuleOp()
     with module.context, ir.Location.unknown():
         transform_module = builtin_d.Module.create()
-        transform_module.operation.attributes["transform.with_named_sequence"] = (
-            ir.UnitAttr.get()
-        )
+        transform_module.operation.attributes[
+            "transform.with_named_sequence"
+        ] = ir.UnitAttr.get()
         with ir.InsertionPoint(transform_module.body):
             named_sequence = NamedSequenceOp("__transform_main", [any_op_t()], [])
             with ir.InsertionPoint(named_sequence.body):