13 #ifndef MLIR_DIALECT_SPARSETENSOR_TRANSFORMS_UTILS_ITERATIONGRAPHSORTER_H_
14 #define MLIR_DIALECT_SPARSETENSOR_TRANSFORMS_UTILS_ITERATIONGRAPHSORTER_H_
24 enum class IteratorType : uint32_t;
30 namespace sparse_tensor {
49 fromGenericOp(linalg::GenericOp genericOp,
58 unsigned getNumLoops()
const {
return loop2OutLvl.getNumDims(); }
89 std::vector<std::vector<bool>> itGraph;
92 std::vector<unsigned> inDegree;
A multi-dimensional affine map Affine map's are immutable like Type's, and they are uniqued.
This class represents an instance of an SSA value in the MLIR system, representing a computable value...
unsigned getNumLoops() const
Returns the number of loops in the iteration graph.
LoopOrderingStrategy
Defines a strategy for loop ordering during sparse code generation.
@ kDefault
Default strategy (eagerly selects last loop in topological sort).
SortMask
Iteration graph sorting mask,.
Include the generated interface declarations.