Skip to content

Commit 0ddd081

Browse files
committed
add more docs and fix some name
1 parent 395627f commit 0ddd081

File tree

3 files changed

+20
-9
lines changed

3 files changed

+20
-9
lines changed

mlir/include/mlir-c/Rewrite.h

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -303,11 +303,15 @@ MLIR_CAPI_EXPORTED void mlirIRRewriterDestroy(MlirRewriterBase rewriter);
303303
/// FrozenRewritePatternSet API
304304
//===----------------------------------------------------------------------===//
305305

306+
/// Freeze the given MlirRewritePatternSet to a MlirFrozenRewritePatternSet.
307+
/// Note that the ownership of the input set is transferred into the frozen set
308+
/// after this call.
306309
MLIR_CAPI_EXPORTED MlirFrozenRewritePatternSet
307-
mlirFreezeRewritePattern(MlirRewritePatternSet op);
310+
mlirFreezeRewritePattern(MlirRewritePatternSet set);
308311

312+
/// Destroy the given MlirFrozenRewritePatternSet.
309313
MLIR_CAPI_EXPORTED void
310-
mlirFrozenRewritePatternSetDestroy(MlirFrozenRewritePatternSet op);
314+
mlirFrozenRewritePatternSetDestroy(MlirFrozenRewritePatternSet set);
311315

312316
MLIR_CAPI_EXPORTED MlirLogicalResult mlirApplyPatternsAndFoldGreedilyWithOp(
313317
MlirOperation op, MlirFrozenRewritePatternSet patterns,
@@ -368,6 +372,8 @@ mlirRewritePatternSetCreate(MlirContext context);
368372
MLIR_CAPI_EXPORTED void mlirRewritePatternSetDestroy(MlirRewritePatternSet set);
369373

370374
/// Add the given MlirRewritePattern into a MlirRewritePatternSet.
375+
/// Note that the ownership of the pattern is transferred to the set after this
376+
/// call.
371377
MLIR_CAPI_EXPORTED void mlirRewritePatternSetAdd(MlirRewritePatternSet set,
372378
MlirRewritePattern pattern);
373379

mlir/lib/Bindings/Python/Rewrite.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,11 @@ class PyRewritePatternSet {
202202
mlirRewritePatternSetAdd(set, pattern);
203203
}
204204

205-
PyFrozenRewritePatternSet freeze() { return mlirFreezeRewritePattern(set); }
205+
PyFrozenRewritePatternSet freeze() {
206+
MlirRewritePatternSet s = set;
207+
set.ptr = nullptr;
208+
return mlirFreezeRewritePattern(s);
209+
}
206210

207211
private:
208212
MlirRewritePatternSet set;

mlir/lib/CAPI/Transforms/Rewrite.cpp

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -291,15 +291,16 @@ wrap(mlir::FrozenRewritePatternSet *module) {
291291
return {module};
292292
}
293293

294-
MlirFrozenRewritePatternSet mlirFreezeRewritePattern(MlirRewritePatternSet op) {
295-
auto *m = new mlir::FrozenRewritePatternSet(std::move(*unwrap(op)));
296-
op.ptr = nullptr;
294+
MlirFrozenRewritePatternSet
295+
mlirFreezeRewritePattern(MlirRewritePatternSet set) {
296+
auto *m = new mlir::FrozenRewritePatternSet(std::move(*unwrap(set)));
297+
set.ptr = nullptr;
297298
return wrap(m);
298299
}
299300

300-
void mlirFrozenRewritePatternSetDestroy(MlirFrozenRewritePatternSet op) {
301-
delete unwrap(op);
302-
op.ptr = nullptr;
301+
void mlirFrozenRewritePatternSetDestroy(MlirFrozenRewritePatternSet set) {
302+
delete unwrap(set);
303+
set.ptr = nullptr;
303304
}
304305

305306
MlirLogicalResult

0 commit comments

Comments
 (0)