snake_case_helper

Author: rolfmorel

mlir/python/mlir/dialects/transform/__init__.py

@@ -249,6 +249,10 @@ def __init__(
         )
 
 
+def apply_registered_pass(result, pass_name, target, *, options=[], loc=None, ip=None) -> Value:
+  return ApplyRegisteredPassOp(result=result, pass_name=pass_name, target=target, options=options, loc=loc, ip=ip).result
+
+
 AnyOpTypeT = NewType("AnyOpType", AnyOpType)
 
 

mlir/test/python/dialects/transform.py

@@ -266,15 +266,18 @@ def testApplyRegisteredPassOp(module: Module):
             transform.AnyOpType.get(), "canonicalize", sequence.bodyTarget
         )
         mod = transform.ApplyRegisteredPassOp(
-            transform.AnyOpType.get(), "canonicalize", mod, options=("top-down=false",)
+            transform.AnyOpType.get(),
+            "canonicalize",
+            mod.result,
+            options=("top-down=false",),
         )
         max_iter = transform.param_constant(
             transform.AnyParamType.get(), StringAttr.get("max-iterations=10")
         )
         max_rewrites = transform.param_constant(
             transform.AnyParamType.get(), StringAttr.get("max-num-rewrites=1")
         )
-        transform.ApplyRegisteredPassOp(
+        transform.apply_registered_pass(
             transform.AnyOpType.get(),
             "canonicalize",
             mod,