Lines Matching defs:rewriter

1 //===- TestPDLByteCode.cpp - Test rewriter bytecode functionality ---------===//
17 static LogicalResult customSingleEntityConstraint(PatternRewriter &rewriter,
21 static LogicalResult customMultiEntityConstraint(PatternRewriter &rewriter,
24   return customSingleEntityConstraint(rewriter, rootCopy);
27     PatternRewriter &rewriter, ValueRange operandValues, TypeRange typeValues) {
34 static LogicalResult customValueResultConstraint(PatternRewriter &rewriter,
39     StringAttr customAttr = rewriter.getStringAttr("test.success");
47 static LogicalResult customTypeResultConstraint(PatternRewriter &rewriter,
52     results.push_back(rewriter.getF32Type());
60 static LogicalResult customTypeRangeResultConstraint(PatternRewriter &rewriter,
69       types.push_back(rewriter.getF32Type());
78 static Operation *customCreate(PatternRewriter &rewriter, Operation *op) {
79   return rewriter.create(OperationState(op->getLoc(), "test.success"));
81 static auto customVariadicResultCreate(PatternRewriter &rewriter,
85 static Type customCreateType(PatternRewriter &rewriter) {
86   return rewriter.getF32Type();
88 static std::string customCreateStrAttr(PatternRewriter &rewriter) {
92 /// Custom rewriter invoked from PDL.
93 static void customRewriter(PatternRewriter &rewriter, Operation *root,
95   rewriter.create(root->getLoc(), rewriter.getStringAttr("test.success"),
97   rewriter.eraseOp(root);
160     pdlPattern.registerRewriteFunction("rewriter", customRewriter);