[all-commits] [llvm/llvm-project] 61a3dd: [mlir][taco] Use sparse_tensor.out to write sparse...

[Bixia Zheng via All-commits]

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

  Changed paths:
    M mlir/test/Integration/Dialect/SparseTensor/taco/data/gold_A.tns
    M mlir/test/Integration/Dialect/SparseTensor/taco/data/gold_C.tns
    M mlir/test/Integration/Dialect/SparseTensor/taco/data/gold_y.tns
    M mlir/test/Integration/Dialect/SparseTensor/taco/test_MTTKRP.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMV.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/mlir_pytaco_utils.py
    A mlir/test/Integration/Dialect/SparseTensor/taco/tools/testing_utils.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/unit_test_tensor_io.py

  Log Message:
  -----------
  [mlir][taco] Use sparse_tensor.out to write sparse tensors to files.

Add a Python method, output_sparse_tensor, to use sparse_tensor.out to write
a sparse tensor value to a file.

Modify the method that evaluates a tensor expression to return a pointer of the
MLIR sparse tensor for the result to delay the extraction of the coordinates and
non-zero values.

Implement the Tensor to_file method to evaluate the tensor assignment and write
the result to a file.

Add unit tests. Modify test golden files to reflect the change that TNS outputs
now have a comment line and two meta data lines.

Reviewed By: aartbik

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