[all-commits] [llvm/llvm-project] 74cd9e: [mlir][sparse] hoist loop invariant tensor loads i...

[Aart Bik via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 74cd9e587d80063381242006d0690231d756aa7a
      https://github.com/llvm/llvm-project/commit/74cd9e587d80063381242006d0690231d756aa7a
  Author: Aart Bik <ajcbik at google.com>
  Date:   2020-12-07 (Mon, 07 Dec 2020)

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

  Log Message:
  -----------
  [mlir][sparse] hoist loop invariant tensor loads in sparse compiler

After bufferization, the backend has much more trouble hoisting loop invariant
loads from the loops generated by the sparse compiler. Therefore, this is done
during sparse code generation. Note that we don't bother hoisting derived
invariant expressions on SSA values, since the backend does that very well.

Still TBD: scalarize reductions to avoid load-add-store cycles

Reviewed By: penpornk

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