1 //===- SparseTensorInterfaces.h - sparse tensor operations interfaces------===// 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_IR_SPARSETENSORINTERFACES_H_ 10 #define MLIR_DIALECT_SPARSETENSOR_IR_SPARSETENSORINTERFACES_H_ 11 12 #include "mlir/IR/OpDefinition.h" 13 14 namespace mlir { 15 class PatternRewriter; 16 17 namespace sparse_tensor { 18 class StageWithSortSparseOp; 19 20 namespace detail { 21 LogicalResult stageWithSortImpl(sparse_tensor::StageWithSortSparseOp op, 22 PatternRewriter &rewriter, Value &tmpBufs); 23 } // namespace detail 24 } // namespace sparse_tensor 25 } // namespace mlir 26 27 /// Include the generated interface declarations. 28 #include "mlir/Dialect/SparseTensor/IR/SparseTensorInterfaces.h.inc" 29 30 #endif // MLIR_DIALECT_SPARSETENSOR_IR_SPARSETENSORINTERFACES_H_ 31