[all-commits] [llvm/llvm-project] 175b9a: [mlir][sparse] avoid reserving dense storage for p...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 175b9af484f483c3423ab2f78db5de7e25b64c31
      https://github.com/llvm/llvm-project/commit/175b9af484f483c3423ab2f78db5de7e25b64c31
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-04-05 (Tue, 05 Apr 2022)

  Changed paths:
    M mlir/lib/ExecutionEngine/SparseTensorUtils.cpp

  Log Message:
  -----------
  [mlir][sparse] avoid reserving dense storage for ptr/idx

This avoids a rather big bug where we were reserving
dense space for the ptx/idx in the first sparse dimension.
For example, using CSR for a 140874 x 140874 matrix with
3977139 nonzero would reserve the full 19845483876 space.
This revision fixes this for now, but we need to revisit
the reservation heuristic to make this better.

Reviewed By: bixia

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