[all-commits] [llvm/llvm-project] ad932a: [mlir][sparse][taco] Support true dense tensors an...

[Bixia Zheng via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: ad932a75f9ae58afbf5d43521d18c9a5a12876ff
      https://github.com/llvm/llvm-project/commit/ad932a75f9ae58afbf5d43521d18c9a5a12876ff
  Author: Bixia Zheng <bixia at google.com>
  Date:   2022-02-14 (Mon, 14 Feb 2022)

  Changed paths:
    M mlir/test/Integration/Dialect/SparseTensor/taco/test_simple_tensor_algebra.py
    A mlir/test/Integration/Dialect/SparseTensor/taco/test_true_dense_tensor_algebra.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_io.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/testing_utils.py
    A mlir/test/Integration/Dialect/SparseTensor/taco/unit_test_tensor_core.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/unit_test_tensor_io.py

  Log Message:
  -----------
  [mlir][sparse][taco] Support true dense tensors and all dense sparse tensors.

The only method to create a true dense tensor (i.e un-annotated) in MLIR-PyTACO
is through the from_array method. However, the annotated all dense tensors are
also implemented as true dense tensor currently. The PR fixes the
implementation to support annotated all dense sparse tensors.

Extend the tensor init method to support the construction of a tensor without
any sparsity annotation.

Change the tensor to_file method to only support writing unpacked sparse
tensors to file through the MLIR sparse tensor dialect.

Add unit tests for true dense tensors and all dense sparse tensors.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D119500