populate with specific patterns

j2kun · j2kun · commit ab63790e45f7 · 2024-05-16T10:26:38.000-07:00
diff --git a/mlir/lib/Dialect/Polynomial/IR/PolynomialOps.cpp b/mlir/lib/Dialect/Polynomial/IR/PolynomialOps.cpp
@@ -196,15 +196,15 @@ namespace {
 
 void SubOp::getCanonicalizationPatterns(RewritePatternSet &results,
                                         MLIRContext *context) {
-  populateWithGenerated(results);
+  results.add<SubAsAdd>(context);
 }
 
 void NTTOp::getCanonicalizationPatterns(RewritePatternSet &results,
                                         MLIRContext *context) {
-  populateWithGenerated(results);
+  results.add<NTTAfterINTT>(context);
 }
 
 void INTTOp::getCanonicalizationPatterns(RewritePatternSet &results,
                                          MLIRContext *context) {
-  populateWithGenerated(results);
+  results.add<INTTAfterNTT>(context);
 }

Original file line number	Diff line number	Diff line change
`@@ -196,15 +196,15 @@ namespace {`
`196`	`196`
`197`	`197`	`void SubOp::getCanonicalizationPatterns(RewritePatternSet &results,`
`198`	`198`	`MLIRContext *context) {`
`199`		`- populateWithGenerated(results);`
	`199`	`+ results.add<SubAsAdd>(context);`
`200`	`200`	`}`
`201`	`201`
`202`	`202`	`void NTTOp::getCanonicalizationPatterns(RewritePatternSet &results,`
`203`	`203`	`MLIRContext *context) {`
`204`		`- populateWithGenerated(results);`
	`204`	`+ results.add<NTTAfterINTT>(context);`
`205`	`205`	`}`
`206`	`206`
`207`	`207`	`void INTTOp::getCanonicalizationPatterns(RewritePatternSet &results,`
`208`	`208`	`MLIRContext *context) {`
`209`		`- populateWithGenerated(results);`
	`209`	`+ results.add<INTTAfterNTT>(context);`
`210`	`210`	`}`