[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