forked from OSchip/llvm-project
334 lines
15 KiB
TableGen
334 lines
15 KiB
TableGen
// 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
|
|
|
|
// RUN: mlir-tblgen -gen-op-defs -I %S/../../include %s | FileCheck %s --check-prefix=DEFS
|
|
|
|
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<(ins "Value":$val)>,
|
|
OpBuilder<(ins CArg<"int", "0">:$integer)>];
|
|
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();
|
|
// CHECK: ::mlir::FloatAttr attr2();
|
|
// CHECK: ::mlir::Region &someRegion();
|
|
// CHECK: ::mlir::RegionRange someRegions();
|
|
// CHECK: private:
|
|
// CHECK: ::mlir::ValueRange odsOperands;
|
|
// CHECK: };
|
|
|
|
// CHECK: class AOp : public ::mlir::Op<AOp, ::mlir::OpTrait::AtLeastNRegions<1>::Impl, ::mlir::OpTrait::AtLeastNResults<1>::Impl, ::mlir::OpTrait::ZeroSuccessor, ::mlir::OpTrait::AtLeastNOperands<1>::Impl, ::mlir::OpTrait::IsIsolatedFromAbove
|
|
// CHECK-NOT: ::mlir::OpTrait::IsIsolatedFromAbove
|
|
// CHECK: public:
|
|
// CHECK: using Op::Op;
|
|
// CHECK: using Adaptor = AOpAdaptor;
|
|
// CHECK: static constexpr ::llvm::StringLiteral getOperationName() {
|
|
// CHECK: return ::llvm::StringLiteral("test.a_op");
|
|
// CHECK: }
|
|
// 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<::mlir::Region> someRegions();
|
|
// CHECK: ::mlir::IntegerAttr attr1Attr()
|
|
// CHECK: uint32_t attr1();
|
|
// CHECK: ::mlir::FloatAttr attr2Attr()
|
|
// CHECK: ::llvm::Optional< ::llvm::APFloat > attr2();
|
|
// CHECK: ::mlir::Attribute removeAttr2Attr();
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, Value val);
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, int integer = 0);
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type r, ::mlir::TypeRange 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, ::mlir::TypeRange s, ::mlir::Value a, ::mlir::ValueRange b, uint32_t attr1, /*optional*/::mlir::FloatAttr attr2, unsigned someRegionsCount)
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange 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::RewritePatternSet &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: };
|
|
|
|
// DEFS-LABEL: NS::AOp definitions
|
|
|
|
// DEFS: AOpAdaptor::AOpAdaptor(::mlir::ValueRange values, ::mlir::DictionaryAttr attrs, ::mlir::RegionRange regions) : odsOperands(values), odsAttrs(attrs), odsRegions(regions)
|
|
// DEFS: ::mlir::RegionRange AOpAdaptor::getRegions()
|
|
// DEFS: ::mlir::RegionRange AOpAdaptor::someRegions()
|
|
// DEFS-NEXT: return odsRegions.drop_front(1);
|
|
|
|
// 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, ::mlir::TypeRange resultTypes, ::mlir::Value a);
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange 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, ::mlir::TypeRange b, ::mlir::ValueRange a);
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange 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, ::mlir::TypeRange b, ::mlir::ValueRange a);
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange 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, ::mlir::TypeRange b, ::mlir::TypeRange c, ::mlir::ValueRange a);
|
|
// CHECK-NOT: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange b, ::mlir::ValueRange a);
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
|
|
|
|
// Check default value of `attributes` for the `genUseOperandAsResultTypeCollectiveParamBuilder` builder
|
|
def NS_IOp : NS_Op<"op_with_same_operands_and_result_types_trait", [SameOperandsAndResultType]> {
|
|
let arguments = (ins AnyType:$a, AnyType:$b);
|
|
let results = (outs AnyType:$r);
|
|
}
|
|
// CHECK_LABEL: class NS_IOp :
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type r, ::mlir::Value a, ::mlir::Value b);
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value a, ::mlir::Value b);
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value a, ::mlir::Value b);
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
|
|
|
|
// Check default value of `attributes` for the `genInferredTypeCollectiveParamBuilder` builder
|
|
def NS_JOp : NS_Op<"op_with_InferTypeOpInterface_interface", [DeclareOpInterfaceMethods<InferTypeOpInterface>]> {
|
|
let arguments = (ins AnyType:$a, AnyType:$b);
|
|
let results = (outs AnyType:$r);
|
|
}
|
|
// CHECK_LABEL: class NS_JOp :
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type r, ::mlir::Value a, ::mlir::Value b);
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value a, ::mlir::Value b);
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value a, ::mlir::Value b);
|
|
// CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
|
|
|
|
// Test usage of OpTraitList getting flattened during emission.
|
|
def NS_KOp : NS_Op<"k_op", [IsolatedFromAbove,
|
|
OpTraitList<[DeclareOpInterfaceMethods<InferTypeOpInterface>]>]> {
|
|
let arguments = (ins AnyType:$a, AnyType:$b);
|
|
}
|
|
|
|
// CHECK: class KOp : public ::mlir::Op<KOp,
|
|
// CHECK-SAME: ::mlir::OpTrait::IsIsolatedFromAbove, ::mlir::InferTypeOpInterface::Trait
|
|
|
|
// Check native OpTrait usage
|
|
// ---
|
|
|
|
def NS_TestTrait : NativeOpTrait<"TestTrait"> {
|
|
let cppNamespace = "SomeNamespace";
|
|
}
|
|
|
|
def NS_KWithTraitOp : NS_Op<"KWithTrait", [NS_TestTrait]>;
|
|
|
|
// CHECK-LABEL: NS::KWithTraitOp declarations
|
|
// CHECK: class KWithTraitOp : public ::mlir::Op<KWithTraitOp
|
|
// CHECK-SAME: SomeNamespace::TestTrait
|
|
|
|
// Test that type defs have the proper namespaces when used as a constraint.
|
|
// ---
|
|
|
|
def Test_Dialect2 : Dialect {
|
|
let name = "test";
|
|
let cppNamespace = "::mlir::dialect2";
|
|
}
|
|
def TestDialect2Type : TypeDef<Test_Dialect2, "Dialect2Type">;
|
|
|
|
def NS_ResultWithDialectTypeOp : NS_Op<"op_with_dialect_type", []> {
|
|
let results = (outs TestDialect2Type);
|
|
}
|
|
|
|
// CHECK-LABEL: NS::ResultWithDialectTypeOp declarations
|
|
// CHECK: class ResultWithDialectTypeOp :
|
|
// CHECK-SAME: ::mlir::OpTrait::OneTypedResult<::mlir::dialect2::Dialect2TypeType>
|
|
|
|
// Check that default builders can be suppressed.
|
|
// ---
|
|
|
|
def NS_SkipDefaultBuildersOp : NS_Op<"skip_default_builders", []> {
|
|
let skipDefaultBuilders = 1;
|
|
let builders = [OpBuilder<(ins "Value":$val)>];
|
|
}
|
|
|
|
// CHECK-LABEL: NS::SkipDefaultBuildersOp declarations
|
|
// CHECK: class SkipDefaultBuildersOp
|
|
// CHECK-NOT: static void build(::mlir::Builder
|
|
// CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, 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
|