[mlir][sparse] first end-to-end matmul with codegen
authorAart Bik <ajcbik@google.com>
Wed, 9 Nov 2022 21:05:43 +0000 (13:05 -0800)
committerAart Bik <ajcbik@google.com>
Wed, 9 Nov 2022 21:40:05 +0000 (13:40 -0800)
commita61a9a700a35ad0def066acd030d98f19c81cabd
treef476ca6e16de8914f91d8c7a0141caf3ff22fde2
parent8bcf5df3043a906c7124b70b59eda925eddd7319
[mlir][sparse] first end-to-end matmul with codegen

(1) also fixes memory leak in sparse2dense rewriting
(2) still needs fix in dense2sparse by skipping zeros

Reviewed By: wrengr

Differential Revision: https://reviews.llvm.org/D137736
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.h
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matmul.mlir