shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
419,937 Commits 1,713 Branches 298 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
Aart Bik
8b83b8f131 [mlir][sparse] refactor sparse compiler pipeline to single place 
Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D120347
 2022-02-22 16:23:56 -08:00
Aart Bik
6438783fda [mlir][sparse] provide more types for external to/from MLIR routines 
These routines will need to be specialized a lot more based on value types,
index types, pointer types, and permutation/dimension ordering. This is a
careful first step, providing some functionality needed in PyTACO bridge.

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D120154
 2022-02-18 13:36:52 -08:00
Shao-Ce SUN
21ac474392 [NFC] Correct typo interger to integer 2022-02-17 21:17:47 +08:00
Bixia Zheng
2f49e6b0db Support sparse tensor output. 
Add convertFromMLIRSparseTensor to the supporting C shared library to convert
SparseTensorStorage to COO-flavor format.

Add Python routine sparse_tensor_to_coo_tensor to convert sparse tensor storage
pointer to numpy values for COO-flavor format tensor.

Add a Python test for sparse tensor output.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D115557
 2021-12-13 12:06:33 -08:00