54 Commits

Author	SHA1	Message	Date
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
Bixia Zheng	64e171c2d0	Avoid unnecessary output buffer allocation and initialization. ... The sparse tensor code generator allocates memory for the output tensor. As such, we only need to allocate a MemRefDescriptor to receive the output tensor and do not need to allocate and initialize the storage for the tensor. Reviewed By: aartbik Differential Revision: https://reviews.llvm.org/D115292	2021-12-09 08:29:02 -08:00
wren romano	4748cc6931	[mlir][sparse] Adding a stress test ... Addresses https://bugs.llvm.org/show_bug.cgi?id=52410 Depends on D114192 Reviewed By: aartbik, mehdi_amini Differential Revision: https://reviews.llvm.org/D114118	2021-12-03 14:59:39 -08:00
wren romano	286248db2c	[mlir][sparse] Moving integration tests that merely use the Python API ... Reviewed By: aartbik Differential Revision: https://reviews.llvm.org/D114192	2021-11-23 10:59:38 -08:00