| // RUN: mlir-tblgen -gen-op-decls -I %S/../../include %s | FileCheck %s |
| // RUN: mlir-tblgen -gen-op-decls -op-include-regex="test.a_op" -I %S/../../include %s | FileCheck %s --check-prefix=REDUCE_INC |
| // RUN: mlir-tblgen -gen-op-decls -op-exclude-regex="test.a_op" -I %S/../../include %s | FileCheck %s --check-prefix=REDUCE_EXC |
| |
| include "mlir/IR/OpBase.td" |
| include "mlir/Interfaces/InferTypeOpInterface.td" |
| include "mlir/Interfaces/SideEffectInterfaces.td" |
| |
| def Test_Dialect : Dialect { |
| let name = "test"; |
| let cppNamespace = "NS"; |
| } |
| class NS_Op<string mnemonic, list<OpTrait> traits> : |
| Op<Test_Dialect, mnemonic, traits>; |
| |
| // IsolatedFromAbove trait is included twice to ensure it gets uniqued during |
| // emission. |
| def NS_AOp : NS_Op<"a_op", [IsolatedFromAbove, IsolatedFromAbove]> { |
| let arguments = (ins |
| I32:$a, |
| Variadic<F32>:$b, |
| |
| I32Attr:$attr1, |
| OptionalAttr<F32Attr>:$attr2 |
| ); |
| |
| let results = (outs |
| I32:$r, |
| Variadic<F32>:$s |
| ); |
| |
| let regions = (region |
| AnyRegion:$someRegion, |
| VariadicRegion<AnyRegion>:$someRegions |
| ); |
| let builders = [OpBuilder<"Value val">]; |
| let parser = [{ foo }]; |
| let printer = [{ bar }]; |
| let verifier = [{ baz }]; |
| |
| let hasCanonicalizer = 1; |
| let hasFolder = 1; |
| |
| let extraClassDeclaration = [{ |
| // Display a graph for debugging purposes. |
| void displayGraph(); |
| }]; |
| } |
| |
| // CHECK-LABEL: NS::AOp declarations |
| |
| // CHECK: class AOpAdaptor { |
| // CHECK: public: |
| // CHECK: AOpAdaptor(::mlir::ValueRange values |
| // CHECK: ::mlir::ValueRange getODSOperands(unsigned index); |
| // CHECK: ::mlir::Value a(); |
| // CHECK: ::mlir::ValueRange b(); |
| // CHECK: ::mlir::IntegerAttr attr1(); |
| // CHECL: ::mlir::FloatAttr attr2(); |
| // CHECK: private: |
| // CHECK: ::mlir::ValueRange odsOperands; |
| // CHECK: }; |
| |
| // CHECK: class AOp : public ::mlir::Op<AOp, OpTrait::AtLeastNRegions<1>::Impl, OpTrait::AtLeastNResults<1>::Impl, OpTrait::ZeroSuccessor, OpTrait::AtLeastNOperands<1>::Impl, OpTrait::IsIsolatedFromAbove |
| // CHECK-NOT: OpTrait::IsIsolatedFromAbove |
| // CHECK: public: |
| // CHECK: using Op::Op; |
| // CHECK: using Adaptor = AOpAdaptor; |
| // CHECK: static ::llvm::StringRef getOperationName(); |
| // CHECK: ::mlir::Operation::operand_range getODSOperands(unsigned index); |
| // CHECK: ::mlir::Value a(); |
| // CHECK: ::mlir::Operation::operand_range b(); |
| // CHECK: ::mlir::MutableOperandRange aMutable(); |
| // CHECK: ::mlir::MutableOperandRange bMutable(); |
| // CHECK: ::mlir::Operation::result_range getODSResults(unsigned index); |
| // CHECK: ::mlir::Value r(); |
| // CHECK: ::mlir::Region &someRegion(); |
| // CHECK: ::mlir::MutableArrayRef<Region> someRegions(); |
| // CHECK: ::mlir::IntegerAttr attr1Attr() |
| // CHECK: ::llvm::APInt attr1(); |
| // CHECK: ::mlir::FloatAttr attr2Attr() |
| // CHECK: ::llvm::Optional< ::llvm::APFloat > attr2(); |
| // CHECK: static void build(Value val); |
| // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type r, ::llvm::ArrayRef<::mlir::Type> s, ::mlir::Value a, ::mlir::ValueRange b, ::mlir::IntegerAttr attr1, /*optional*/::mlir::FloatAttr attr2, unsigned someRegionsCount) |
| // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type r, ::llvm::ArrayRef<::mlir::Type> s, ::mlir::Value a, ::mlir::ValueRange b, ::llvm::APInt attr1, /*optional*/::mlir::FloatAttr attr2, unsigned someRegionsCount) |
| // CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes, unsigned numRegions) |
| // CHECK: static ::mlir::ParseResult parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result); |
| // CHECK: void print(::mlir::OpAsmPrinter &p); |
| // CHECK: ::mlir::LogicalResult verify(); |
| // CHECK: static void getCanonicalizationPatterns(::mlir::OwningRewritePatternList &results, ::mlir::MLIRContext *context); |
| // CHECK: ::mlir::LogicalResult fold(::llvm::ArrayRef<::mlir::Attribute> operands, ::llvm::SmallVectorImpl<::mlir::OpFoldResult> &results); |
| // CHECK: // Display a graph for debugging purposes. |
| // CHECK: void displayGraph(); |
| // CHECK: }; |
| |
| // Check AttrSizedOperandSegments |
| // --- |
| |
| def NS_AttrSizedOperandOp : NS_Op<"attr_sized_operands", |
| [AttrSizedOperandSegments]> { |
| let arguments = (ins |
| Variadic<I32>:$a, |
| Variadic<I32>:$b, |
| I32:$c, |
| Variadic<I32>:$d, |
| I32ElementsAttr:$operand_segment_sizes |
| ); |
| } |
| |
| // CHECK-LABEL: AttrSizedOperandOpAdaptor( |
| // CHECK-SAME: ::mlir::ValueRange values |
| // CHECK-SAME: ::mlir::DictionaryAttr attrs |
| // CHECK: ::mlir::ValueRange a(); |
| // CHECK: ::mlir::ValueRange b(); |
| // CHECK: ::mlir::Value c(); |
| // CHECK: ::mlir::ValueRange d(); |
| // CHECK: ::mlir::DenseIntElementsAttr operand_segment_sizes(); |
| |
| // Check op trait for different number of operands |
| // --- |
| |
| def NS_BOp : NS_Op<"op_with_no_operand", []> { |
| let arguments = (ins); |
| } |
| |
| // CHECK-LABEL: NS::BOp declarations |
| // CHECK: OpTrait::ZeroOperands |
| |
| def NS_COp : NS_Op<"op_with_one_operand", []> { |
| let arguments = (ins I32:$operand); |
| } |
| |
| // CHECK-LABEL: NS::COp declarations |
| // CHECK: OpTrait::OneOperand |
| |
| def NS_DOp : NS_Op<"op_with_two_operands", []> { |
| let arguments = (ins I32:$input1, I32:$input2); |
| } |
| |
| // CHECK-LABEL: NS::DOp declarations |
| // CHECK: OpTrait::NOperands<2>::Impl |
| |
| def NS_EOp : NS_Op<"op_with_optionals", []> { |
| let arguments = (ins Optional<I32>:$a); |
| let results = (outs Optional<F32>:$b); |
| } |
| |
| // CHECK-LABEL: NS::EOp declarations |
| // CHECK: ::mlir::Value a(); |
| // CHECK: ::mlir::MutableOperandRange aMutable(); |
| // CHECK: ::mlir::Value b(); |
| // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, /*optional*/::mlir::Type b, /*optional*/::mlir::Value a) |
| |
| // Check that all types match constraint results in generating builder. |
| // --- |
| |
| def NS_FOp : NS_Op<"op_with_all_types_constraint", |
| [AllTypesMatch<["a", "b"]>]> { |
| let arguments = (ins AnyType:$a); |
| let results = (outs Res<AnyType, "output b", []>:$b); |
| } |
| |
| // CHECK-LABEL: class FOp : |
| // CHECK: static ::mlir::LogicalResult inferReturnTypes |
| |
| def NS_GOp : NS_Op<"op_with_fixed_return_type", []> { |
| let arguments = (ins AnyType:$a); |
| let results = (outs I32:$b); |
| } |
| |
| // CHECK-LABEL: class GOp : |
| // CHECK: static ::mlir::LogicalResult inferReturnTypes |
| |
| // Check default value for collective params builder. Check that other builders |
| // are generated as well. |
| def NS_HCollectiveParamsOp : NS_Op<"op_collective_params", []> { |
| let arguments = (ins AnyType:$a); |
| let results = (outs AnyType:$b); |
| } |
| |
| // CHECK_LABEL: class NS_HCollectiveParamsOp : |
| // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type b, ::mlir::Value a); |
| // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> resultTypes, ::mlir::Value a); |
| // CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {}) |
| |
| // Check suppression of "separate arg, separate result" build method for an op |
| // with single variadic arg and single variadic result (since it will be |
| // ambiguous with the collective params build method). |
| def NS_HCollectiveParamsSuppress0Op : NS_Op<"op_collective_suppress0", []> { |
| let arguments = (ins Variadic<I32>:$a); |
| let results = (outs Variadic<I32>:$b); |
| } |
| |
| // CHECK_LABEL: class NS_HCollectiveParamsSuppress0Op : |
| // CHECK-NOT: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> b, ::mlir::ValueRange a); |
| // CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {}); |
| |
| // Check suppression of "separate arg, collective result" build method for an op |
| // with single variadic arg and non variadic result (since it will be |
| // ambiguous with the collective params build method). |
| def NS_HCollectiveParamsSuppress1Op : NS_Op<"op_collective_suppress1", []> { |
| let arguments = (ins Variadic<I32>:$a); |
| let results = (outs I32:$b); |
| } |
| |
| // CHECK_LABEL: class NS_HCollectiveParamsSuppress1Op : |
| // CHECK-NOT: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> b, ::mlir::ValueRange a); |
| // CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {}); |
| |
| // Check suppression of "separate arg, collective result" build method for an op |
| // with single variadic arg and > 1 variadic result (since it will be |
| // ambiguous with the collective params build method). Note that "separate arg, |
| // separate result" build method should be generated in this case as its not |
| // ambiguous with the collective params build method. |
| def NS_HCollectiveParamsSuppress2Op : NS_Op<"op_collective_suppress2", [SameVariadicResultSize]> { |
| let arguments = (ins Variadic<I32>:$a); |
| let results = (outs Variadic<I32>:$b, Variadic<F32>:$c); |
| } |
| // CHECK_LABEL: class NS_HCollectiveParamsSuppress2Op : |
| // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> b, ::llvm::ArrayRef<::mlir::Type> c, ::mlir::ValueRange a); |
| // CHECK-NOT: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> b, ::mlir::ValueRange a); |
| // CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::llvm::ArrayRef<::mlir::Type> resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {}); |
| |
| // Check that default builders can be suppressed. |
| // --- |
| |
| def NS_SkipDefaultBuildersOp : NS_Op<"skip_default_builders", []> { |
| let skipDefaultBuilders = 1; |
| let builders = [OpBuilder<"Value val">]; |
| } |
| |
| // CHECK-LABEL: NS::SkipDefaultBuildersOp declarations |
| // CHECK: class SkipDefaultBuildersOp |
| // CHECK-NOT: static void build(::mlir::Builder |
| // CHECK: static void build(Value |
| |
| // Check leading underscore in op name |
| // --- |
| |
| def NS__AOp : NS_Op<"_op_with_leading_underscore", []>; |
| |
| // CHECK-LABEL: NS::_AOp declarations |
| // CHECK: class _AOp : public ::mlir::Op<_AOp |
| |
| def _BOp : NS_Op<"_op_with_leading_underscore_and_no_namespace", []>; |
| |
| // CHECK-LABEL: _BOp declarations |
| // CHECK: class _BOp : public ::mlir::Op<_BOp |
| |
| // REDUCE_INC-LABEL: NS::AOp declarations |
| // REDUCE_INC-NOT: NS::BOp declarations |
| |
| // REDUCE_EXC-NOT: NS::AOp declarations |
| // REDUCE_EXC-LABEL: NS::BOp declarations |
