[all-commits] [llvm/llvm-project] af4255: [mlir][sparse] refine optimization, add few more t...

[Aart Bik via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: af42550523d9feef72e6838d5b4e553ef4af79f9
      https://github.com/llvm/llvm-project/commit/af42550523d9feef72e6838d5b4e553ef4af79f9
  Author: Aart Bik <ajcbik at google.com>
  Date:   2020-11-20 (Fri, 20 Nov 2020)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Sparsification.cpp
    M mlir/test/Dialect/Linalg/sparse_1d.mlir
    M mlir/test/Dialect/Linalg/sparse_2d.mlir
    M mlir/test/Dialect/Linalg/sparse_3d.mlir

  Log Message:
  -----------
  [mlir][sparse] refine optimization, add few more test cases

Adds tests for full sum reduction (tensors summed up into scalars)
and the well-known sampled-dense-dense-matrix-product. Refines
the optimizations rules slightly to handle the summation better.

Reviewed By: penpornk

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