Skip to content

Commit 3cc8cfe

Browse files
committed
remove pattern parameter in callback
1 parent 14d1fb4 commit 3cc8cfe

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

mlir/lib/Bindings/Python/Rewrite.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,11 +193,11 @@ class PyRewritePatternSet {
193193
callbacks.destruct = [](void *userData) {
194194
nb::handle(static_cast<PyObject *>(userData)).dec_ref();
195195
};
196-
callbacks.matchAndRewrite = [](MlirRewritePattern pattern, MlirOperation op,
196+
callbacks.matchAndRewrite = [](MlirRewritePattern, MlirOperation op,
197197
MlirPatternRewriter rewriter,
198198
void *userData) -> MlirLogicalResult {
199199
nb::handle f(static_cast<PyObject *>(userData));
200-
nb::object res = f(op, PyPatternRewriter(rewriter), pattern);
200+
nb::object res = f(op, PyPatternRewriter(rewriter));
201201
return logicalResultFromObject(res);
202202
};
203203
MlirRewritePattern pattern = mlirOpRewritePattenCreate(

mlir/test/python/rewrite.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,12 @@ def run(f):
1515
# CHECK-LABEL: TEST: testRewritePattern
1616
@run
1717
def testRewritePattern():
18-
def to_muli(op, rewriter, pattern):
18+
def to_muli(op, rewriter):
1919
with rewriter.ip:
2020
new_op = arith.muli(op.operands[0], op.operands[1], loc=op.location)
2121
rewriter.replace_op(op, new_op.owner)
2222

23-
def constant_1_to_2(op, rewriter, pattern):
23+
def constant_1_to_2(op, rewriter):
2424
c = op.attributes["value"].value
2525
if c != 1:
2626
return True # failed to match

0 commit comments

Comments
 (0)