MLIR  19.0.0git
SparseTensorInterfaces.h
Go to the documentation of this file.
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_
A special type of RewriterBase that coordinates the application of a rewrite pattern on the current I...
Definition: PatternMatch.h:785
This class represents an instance of an SSA value in the MLIR system, representing a computable value...
Definition: Value.h:96
LogicalResult stageWithSortImpl(sparse_tensor::StageWithSortSparseOp op, PatternRewriter &rewriter, Value &tmpBufs)
Include the generated interface declarations.