[all-commits] [llvm/llvm-project] c25f3d: [mlir][sparse][taco] Support tensor dimension stor...
Bixia Zheng via All-commits
all-commits at lists.llvm.org
Tue Mar 1 15:36:51 PST 2022
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: c25f3dfff3978db8c7728e094e2dd61ab8db5ec5
https://github.com/llvm/llvm-project/commit/c25f3dfff3978db8c7728e094e2dd61ab8db5ec5
Author: Bixia Zheng <bixia at google.com>
Date: 2022-03-01 (Tue, 01 Mar 2022)
Changed paths:
M mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py
M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py
Log Message:
-----------
[mlir][sparse][taco] Support tensor dimension storage ordering and more general
sparsity values.
Previously, we can't properly handle input tensors with a dimension
ordering that is different from the natural ordering or with a mixed of
compressed and dense dimensions. This change fixes the problems by
passing the dimension ordering and sparsity values to the runtime
routine.
Modify an existing test to test the situation.
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D120777
More information about the All-commits
mailing list