40 lines · c
1//===- TestTransformDialectExtension.h --------------------------*- C++ -*-===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// This file defines an extension of the MLIR Transform dialect for testing10// purposes.11//12//===----------------------------------------------------------------------===//13 14#ifndef MLIR_TESTTRANSFORMDIALECTEXTENSION_H15#define MLIR_TESTTRANSFORMDIALECTEXTENSION_H16 17#include "mlir/Bytecode/BytecodeOpInterface.h"18#include "mlir/Dialect/PDL/IR/PDLTypes.h"19#include "mlir/Dialect/Transform/IR/TransformTypes.h"20#include "mlir/Dialect/Transform/Interfaces/MatchInterfaces.h"21#include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"22#include "mlir/IR/OpImplementation.h"23 24namespace mlir {25class DialectRegistry;26} // namespace mlir27 28#define GET_TYPEDEF_CLASSES29#include "TestTransformDialectExtensionTypes.h.inc"30 31#define GET_OP_CLASSES32#include "TestTransformDialectExtension.h.inc"33 34namespace test {35/// Registers the test extension to the Transform dialect.36void registerTestTransformDialectExtension(::mlir::DialectRegistry ®istry);37} // namespace test38 39#endif // MLIR_TESTTRANSFORMDIALECTEXTENSION_H40