[all-commits] [llvm/llvm-project] 4f2ec7: [mlir][sparse] finalize sparse output in the prese...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 4f2ec7f983b40c5796beff96b0bc846a9dacac25
      https://github.com/llvm/llvm-project/commit/4f2ec7f983b40c5796beff96b0bc846a9dacac25
  Author: Aart Bik <ajcbik at google.com>
  Date:   2021-12-07 (Tue, 07 Dec 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
    M mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
    M mlir/lib/ExecutionEngine/SparseTensorUtils.cpp
    M mlir/test/Dialect/SparseTensor/conversion.mlir
    M mlir/test/Dialect/SparseTensor/invalid.mlir
    M mlir/test/Dialect/SparseTensor/roundtrip.mlir
    A mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matmul.mlir

  Log Message:
  -----------
  [mlir][sparse] finalize sparse output in the presence of reductions

This revision implements sparse outputs (from scratch) in all cases where
the loops can be reordered with all but one parallel loops outer. If the
inner parallel loop appears inside one or more reductions loops, then an
access pattern expansion is required (aka. workspaces in TACO speak).

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D115091