[Mlir-commits] [mlir] ff57f40 - [mlir][py] fix option passing in transform interpreter (#89922)

[llvmlistbot at llvm.org]

Author: Oleksandr "Alex" Zinenko
Date: 2024-04-24T19:40:53+02:00
New Revision: ff57f40673f0db2c1a867e5697d5407bc9f39a5e

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

LOG: [mlir][py] fix option passing in transform interpreter (#89922)

There was a typo in dispatch trampoline.

Added: 
    

Modified: 
    mlir/python/mlir/dialects/transform/interpreter/__init__.py
    mlir/test/python/dialects/transform_interpreter.py

Removed: 
    


################################################################################
diff  --git a/mlir/python/mlir/dialects/transform/interpreter/__init__.py b/mlir/python/mlir/dialects/transform/interpreter/__init__.py
index 34cdc43cb617fd..e69aa963038f23 100644
--- a/mlir/python/mlir/dialects/transform/interpreter/__init__.py
+++ b/mlir/python/mlir/dialects/transform/interpreter/__init__.py
@@ -29,7 +29,7 @@ def apply_named_sequence(
     if transform_options is None:
         _cextTransformInterpreter.apply_named_sequence(*args)
     else:
-        _cextTransformInterpreter(*args, transform_options)
+        _cextTransformInterpreter.apply_named_sequence(*args, transform_options)
 
 
 def copy_symbols_and_merge_into(target, other):

diff  --git a/mlir/test/python/dialects/transform_interpreter.py b/mlir/test/python/dialects/transform_interpreter.py
index 807a98c4932797..819a3be1db9d5a 100644
--- a/mlir/test/python/dialects/transform_interpreter.py
+++ b/mlir/test/python/dialects/transform_interpreter.py
@@ -45,6 +45,21 @@ def print_other():
 # CHECK: this.is.payload
 
 
+ at test_in_context
+def transform_options():
+    options = interp.TransformOptions()
+    options.expensive_checks = False
+    options.enforce_single_top_level_transform_op = True
+    m = ir.Module.parse(
+        print_root_module.replace("from interpreter", "transform_options")
+    )
+    payload = ir.Module.parse("module attributes { this.is.payload } {}")
+    interp.apply_named_sequence(payload, m.body.operations[0], m, options)
+
+
+# CHECK-LABEL: transform_options
+
+
 @test_in_context
 def failed():
     payload = ir.Module.parse("module attributes { this.is.payload } {}")