[all-commits] [llvm/llvm-project] 76a186: [mlir][SparseTensor] Split scf.for loop into maske...

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 76a186181634feaeaa2d0493aac2b796d2a3ef25
      https://github.com/llvm/llvm-project/commit/76a186181634feaeaa2d0493aac2b796d2a3ef25
  Author: Matthias Springer <springerm at google.com>
  Date:   2021-08-19 (Thu, 19 Aug 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.td
    M mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
    M mlir/test/Dialect/SparseTensor/sparse_vector.mlir
    A mlir/test/Dialect/SparseTensor/sparse_vector_peeled.mlir
    M utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

  Log Message:
  -----------
  [mlir][SparseTensor] Split scf.for loop into masked/unmasked parts

Apply the "for loop peeling" pattern from SCF dialect transforms. This pattern splits scf.for loops into full and partial iterations. In the full iteration, all masked loads/stores are canonicalized to unmasked loads/stores.

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