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-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file defines an extension of the MLIR Transform dialect for testing 10 // purposes. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef MLIR_TESTTRANSFORMDIALECTEXTENSION_H 15 #define MLIR_TESTTRANSFORMDIALECTEXTENSION_H 16 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 24 namespace mlir { 25 class DialectRegistry; 26 } // namespace mlir 27 28 #define GET_TYPEDEF_CLASSES 29 #include "TestTransformDialectExtensionTypes.h.inc" 30 31 #define GET_OP_CLASSES 32 #include "TestTransformDialectExtension.h.inc" 33 34 namespace test { 35 /// Registers the test extension to the Transform dialect. 36 void registerTestTransformDialectExtension(::mlir::DialectRegistry ®istry); 37 } // namespace test 38 39 #endif // MLIR_TESTTRANSFORMDIALECTEXTENSION_H 40