[all-commits] [llvm/llvm-project] 20eaa8: [mlir][sparse] Extend convertToMLIRSparseTensor to...
Bixia Zheng via All-commits
all-commits at lists.llvm.org
Tue Mar 1 10:51:57 PST 2022
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: 20eaa88fff478616259668c9917332d1f505ccdd
https://github.com/llvm/llvm-project/commit/20eaa88fff478616259668c9917332d1f505ccdd
Author: Bixia Zheng <bixia at google.com>
Date: 2022-03-01 (Tue, 01 Mar 2022)
Changed paths:
M mlir/lib/ExecutionEngine/SparseTensorUtils.cpp
M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py
Log Message:
-----------
[mlir][sparse] Extend convertToMLIRSparseTensor to support permutation and more general sparsity values.
Previously, convertToMLIRSparseTensor assumes identity storage ordering and all
compressed dimensions. This change extends the function with two parameters for
users to specify the storage ordering and the sparsity of each dimension.
Modify PyTACO to reflect this change.
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D120643
More information about the All-commits
mailing list