[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