forked from OSchip/llvm-project
32 lines
1.2 KiB
TableGen
32 lines
1.2 KiB
TableGen
// RUN: mlir-tblgen -gen-rewriters -I %S/../../include %s | FileCheck %s
|
|
|
|
include "mlir/IR/OpBase.td"
|
|
|
|
// Create a Type and Attribute.
|
|
def T : BuildableType<"buildT">;
|
|
def T_Attr : TypeBasedAttr<T, "Attribute", "attribute of T type">;
|
|
def T_Const_Attr : ConstantAttr<T_Attr, "attrValue">;
|
|
|
|
// Define ops to rewrite.
|
|
def U: Type<CPred<"true">, "U">;
|
|
def X_AddOp : Op<"x.add"> {
|
|
let arguments = (ins U, U);
|
|
}
|
|
def Y_AddOp : Op<"y.add"> {
|
|
let arguments = (ins U, U, T_Attr:$attrName);
|
|
}
|
|
|
|
// Define rewrite pattern.
|
|
def : Pat<(X_AddOp:$res $lhs, $rhs), (Y_AddOp $lhs, U:$rhs, T_Const_Attr:$x)>;
|
|
def : Pat<(X_AddOp (X_AddOp $lhs, $rhs):$res, $rrhs), (Y_AddOp $lhs, U:$rhs, T_Const_Attr:$x)>;
|
|
def : Pat<(X_AddOp (X_AddOp:$res $lhs, $rhs), $rrhs), (Y_AddOp $lhs, U:$rhs, T_Const_Attr:$x)>;
|
|
|
|
// CHECK: struct GeneratedConvert0 : public RewritePattern
|
|
// CHECK: RewritePattern("x.add", 1, context)
|
|
// CHECK: PatternMatchResult match(Instruction *
|
|
// CHECK: void rewrite(Instruction *op, std::unique_ptr<PatternState>
|
|
// CHECK: PatternRewriter &rewriter)
|
|
// CHECK: rewriter.create<Y::AddOp>(loc, op->getResult(0)->getType()
|
|
// CHECK: void populateWithGenerated
|
|
// CHECK: patterns->push_back(llvm::make_unique<GeneratedConvert0>(context))
|