[all-commits] [llvm/llvm-project] 0b55f9: [mlir][sparse] replace stack-based access pattern ...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0b55f94d2bf3b0de027e6e674c99210d20bd8f7b
      https://github.com/llvm/llvm-project/commit/0b55f94d2bf3b0de027e6e674c99210d20bd8f7b
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-04-06 (Wed, 06 Apr 2022)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
    M mlir/test/Dialect/SparseTensor/conversion.mlir
    A mlir/test/Dialect/SparseTensor/sparse_expand.mlir

  Log Message:
  -----------
  [mlir][sparse] replace stack-based access pattern with dyn-alloc

Rationale:
Allocating the temporary buffers for access pattern expansion on the stack
(using alloca) is a bit too agressive, since it easily runs out of stack space
for large enveloping tensor dimensions. This revision changes the dynamic
allocation of these buffers with explicit alloc/dealloc pairs.

Reviewed By: bixia, wrengr

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