diff options
Diffstat (limited to 'mlir/test/lib/TestDialect/TestDialect.h')
-rw-r--r-- | mlir/test/lib/TestDialect/TestDialect.h | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/mlir/test/lib/TestDialect/TestDialect.h b/mlir/test/lib/TestDialect/TestDialect.h new file mode 100644 index 00000000000..20db0f39b81 --- /dev/null +++ b/mlir/test/lib/TestDialect/TestDialect.h @@ -0,0 +1,53 @@ +//===- TestDialect.h - MLIR Dialect for testing -----------------*- C++ -*-===// +// +// Part of the MLIR Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// This file defines a fake 'test' dialect that can be used for testing things +// that do not have a respective counterpart in the main source directories. +// +//===----------------------------------------------------------------------===// + +#ifndef MLIR_TESTDIALECT_H +#define MLIR_TESTDIALECT_H + +#include "mlir/Analysis/CallInterfaces.h" +#include "mlir/Analysis/InferTypeOpInterface.h" +#include "mlir/Dialect/Traits.h" +#include "mlir/IR/Dialect.h" +#include "mlir/IR/OpDefinition.h" +#include "mlir/IR/OpImplementation.h" +#include "mlir/IR/StandardTypes.h" +#include "mlir/IR/SymbolTable.h" + +#include "TestOpEnums.h.inc" + +namespace mlir { + +class TestDialect : public Dialect { +public: + /// Create the dialect in the given `context`. + TestDialect(MLIRContext *context); + + /// Get the canonical string name of the dialect. + static StringRef getDialectName() { return "test"; } + + LogicalResult verifyOperationAttribute(Operation *op, + NamedAttribute namedAttr) override; + LogicalResult verifyRegionArgAttribute(Operation *op, unsigned regionIndex, + unsigned argIndex, + NamedAttribute namedAttr) override; + LogicalResult verifyRegionResultAttribute(Operation *op, unsigned regionIndex, + unsigned resultIndex, + NamedAttribute namedAttr) override; +}; + +#define GET_OP_CLASSES +#include "TestOps.h.inc" + +} // end namespace mlir + +#endif // MLIR_TESTDIALECT_H |