llvm-project/mlir/lib/Dialect/SparseTensor
Aart Bik 7373cabcda [mlir][sparse] implement full reduction "scalarization" across loop nests
The earlier reduction "scalarization" was only applied to a chain of
*innermost* and *for* loops. This revision generalizes this to any
nesting of for- and while-loops. This implies that reductions can be
implemented with a lot less load and store operations. The chaining
is implemented with a forest of yield statements (but not as bad as
when we would also include the while-induction).

Fixes https://bugs.llvm.org/show_bug.cgi?id=52311

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D113078
2021-11-04 17:38:47 -07:00
..
IR [mlir][sparse] reject sparsity annotation in "scalar" tensors 2021-11-04 09:49:05 -07:00
Transforms [mlir][sparse] implement full reduction "scalarization" across loop nests 2021-11-04 17:38:47 -07:00
Utils [MLIR] Replace std ops with arith dialect ops 2021-10-13 03:07:03 +00:00
CMakeLists.txt [MLIR][Sparse] Refactor lattice code into its own file 2021-06-24 23:03:44 +00:00