1 //===- NVGPUTransformOps.h - NVGPU transform ops ----------------*- 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 #ifndef MLIR_DIALECT_NVGPU_TRANSFORMOPS_NVGPUTRANSFORMOPS_H 10 #define MLIR_DIALECT_NVGPU_TRANSFORMOPS_NVGPUTRANSFORMOPS_H 11 12 #include "mlir/Dialect/Transform/IR/TransformAttrs.h" 13 #include "mlir/Dialect/Transform/IR/TransformDialect.h" 14 #include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h" 15 #include "mlir/IR/OpImplementation.h" 16 #include "mlir/IR/RegionKindInterface.h" 17 18 namespace mlir { 19 namespace transform { 20 class TransformHandleTypeInterface; 21 } // namespace transform 22 } // namespace mlir 23 24 namespace mlir { 25 class DialectRegistry; 26 27 namespace linalg { 28 class LinalgOp; 29 } // namespace linalg 30 31 namespace scf { 32 class ForOp; 33 } // namespace scf 34 35 namespace nvgpu { 36 void registerTransformDialectExtension(DialectRegistry ®istry); 37 } // namespace nvgpu 38 } // namespace mlir 39 40 //===----------------------------------------------------------------------===// 41 // NVGPU Transform Operations 42 //===----------------------------------------------------------------------===// 43 44 #define GET_OP_CLASSES 45 #include "mlir/Dialect/NVGPU/TransformOps/NVGPUTransformOps.h.inc" 46 47 #endif // MLIR_DIALECT_NVGPU_TRANSFORMOPS_NVGPUTRANSFORMOPS_H 48