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/Dialect/PDL/IR/PDLTypes.h"
18 #include "mlir/Dialect/Transform/IR/TransformInterfaces.h"
19 #include "mlir/IR/OpImplementation.h"
20 
21 namespace mlir {
22 class DialectRegistry;
23 } // namespace mlir
24 
25 #define GET_OP_CLASSES
26 #include "TestTransformDialectExtension.h.inc"
27 
28 namespace test {
29 /// Registers the test extension to the Transform dialect.
30 void registerTestTransformDialectExtension(::mlir::DialectRegistry &registry);
31 } // namespace test
32 
33 #endif // MLIR_TESTTRANSFORMDIALECTEXTENSION_H
34