We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a98ae8 commit 751fe84Copy full SHA for 751fe84
mlir/test/python/pass.py
@@ -55,22 +55,25 @@ def testCustomPass():
55
class CustomPass(Pass):
56
def __init__(self):
57
super().__init__("CustomPass", op_name="builtin.module")
58
+
59
def run(self, m):
60
frozen = PDLModule(pdl_module).freeze()
61
apply_patterns_and_fold_greedily_for_op(m, frozen)
62
- module = ModuleOp.parse(r"""
63
+ module = ModuleOp.parse(
64
+ r"""
65
module {
66
func.func @add(%a: index, %b: index) -> index {
67
%sum = arith.addi %a, %b : index
68
return %sum : index
69
}
70
- """)
71
+ """
72
+ )
73
74
# CHECK-LABEL: Dump After CustomPass
75
# CHECK: arith.muli
- pm = PassManager('any')
76
+ pm = PassManager("any")
77
pm.enable_ir_printing()
78
pm.add(CustomPass())
79
pm.run(module)
0 commit comments