[all-commits] [llvm/llvm-project] 2a2886: [mlir][sparse] improved testing and codegen for se...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2a2886160d80f285d7637f91133ded7bac0f9879
      https://github.com/llvm/llvm-project/commit/2a2886160d80f285d7637f91133ded7bac0f9879
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-06-16 (Thu, 16 Jun 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
    M mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
    M mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
    A mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_triangular_bin.mlir

  Log Message:
  -----------
  [mlir][sparse] improved testing and codegen for semi-ring operations

The semi-ring blocks were simply "inlined" by the sparse compiler but
without any filtering or patching. This revision improves the analysis
(rejecting blocks that use non-invariant computations from outside
their blocks, except for linalg.index) and also improves the codegen
by properly patching up index computations (previous version crashed).

With a regression test. Also updated the documentation now that the
example code is properly working.

Reviewed By: bixia

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