1 //===- SparseTensorTransformOps.h - sparse tensor 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_SPARSETENSOR_TRANSFORMOPS_SPARSETENSORTRANSFORMOPS_H 10 #define MLIR_DIALECT_SPARSETENSOR_TRANSFORMOPS_SPARSETENSORTRANSFORMOPS_H 11 12 #include "mlir/Dialect/Transform/IR/TransformAttrs.h" 13 #include "mlir/Dialect/Transform/IR/TransformDialect.h" 14 #include "mlir/Dialect/Transform/Interfaces/MatchInterfaces.h" 15 #include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h" 16 #include "mlir/IR/OpImplementation.h" 17 #include "mlir/IR/RegionKindInterface.h" 18 19 namespace mlir { 20 namespace transform { 21 class TransformHandleTypeInterface; 22 } // namespace transform 23 } // namespace mlir 24 25 namespace mlir { 26 class DialectRegistry; 27 28 namespace sparse_tensor { 29 void registerTransformDialectExtension(DialectRegistry ®istry); 30 } // namespace sparse_tensor 31 } // namespace mlir 32 33 //===----------------------------------------------------------------------===// 34 // SparseTensor Transform Operations 35 //===----------------------------------------------------------------------===// 36 37 #define GET_OP_CLASSES 38 #include "mlir/Dialect/SparseTensor/TransformOps/SparseTensorTransformOps.h.inc" 39 40 #endif // MLIR_DIALECT_SPARSETENSOR_TRANSFORMOPS_SPARSETENSORTRANSFORMOPS_H 41