[all-commits] [llvm/llvm-project] 03fe15: [mlir][sparse] speed up sparse tensor file I/O by ...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 03fe15cee6e2aa1444f955c86aadfed529353831
      https://github.com/llvm/llvm-project/commit/03fe15cee6e2aa1444f955c86aadfed529353831
  Author: Aart Bik <ajcbik at google.com>
  Date:   2021-12-14 (Tue, 14 Dec 2021)

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

  Log Message:
  -----------
  [mlir][sparse] speed up sparse tensor file I/O by more than 2x

data point using the 3-dim tensor nell-2.tns

MLIR:
READ FILE INTO COO: 24424.369294 ms ---> improves to ----> 9638.501044 ms
SORT COO BEFORE PACK: 762.834831 ms
PACK COO TO TENSOR: 1243.376245 ms

TACO:
b file read: 13270.9 ms
b pack: 7137.74 ms
b size: (12092 x 9184 x 28818), 925300328 bytes

https://github.com/llvm/llvm-project/issues/52679

Reviewed By: bixia

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