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

[llvmlistbot at llvm.org]

Author: Maksim Levental
Date: 2026-01-16T15:50:58Z
New Revision: 2c16364d75e56379d45cc0a26ca92ba56c0662a2

URL: https://github.com/llvm/llvm-project/commit/2c16364d75e56379d45cc0a26ca92ba56c0662a2
DIFF: https://github.com/llvm/llvm-project/commit/2c16364d75e56379d45cc0a26ca92ba56c0662a2.diff

LOG: [MLIR][Python] add builtin module transform test (#176388)

See https://github.com/llvm/llvm-project/pull/176299

Added: 
    

Modified: 
    mlir/test/python/dialects/transform_interpreter.py

Removed: 
    


################################################################################
diff  --git a/mlir/test/python/dialects/transform_interpreter.py b/mlir/test/python/dialects/transform_interpreter.py
index ca9ce5dbd23c1..491afb3f8bd5a 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):
@@ -159,3 +165,24 @@ 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([])
+        transform_module.operation.verify()
+        interp.apply_named_sequence(
+            module,
+            transform_module.body.operations[0],
+            transform_module,
+        )
+        module.operation.verify()