[all-commits] [llvm/llvm-project] 6d8e2f: [mlir][sparse] implement simple reshaping (expand/...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 6d8e2f1e51776a46d2492054f0d702675ff83027
      https://github.com/llvm/llvm-project/commit/6d8e2f1e51776a46d2492054f0d702675ff83027
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-07-06 (Wed, 06 Jul 2022)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
    A mlir/test/Dialect/SparseTensor/rewriting.mlir
    A mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_reshape.mlir

  Log Message:
  -----------
  [mlir][sparse] implement simple reshaping (expand/collapse)

The revision makes a start with implementing expand/collapse reshaping
for sparse tensors. When either source or destination is sparse, but
other is dense, the "cheap" dense reshape can be used prior to converting
from or to a sparse tensor.

Note1
sparse to sparse reshaping is still TBD.

Note2
in the long run, we may want to implement a "view" into a sparse tensor so that the operation remains cheap and does not require data shuffling

Reviewed By: wrengr

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