[all-commits] [llvm/llvm-project] 93c81f: [mlir][taco] Uses sparse_tensor.new to read tensor...

Bixia Zheng via All-commits all-commits at lists.llvm.org
Thu Feb 3 08:26:46 PST 2022


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

  Changed paths:
    M mlir/test/Integration/Dialect/SparseTensor/taco/data/nell-2.tns
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_io.py
    A mlir/test/Integration/Dialect/SparseTensor/taco/unit_test_tensor_io.py

  Log Message:
  -----------
  [mlir][taco] Uses sparse_tensor.new to read tensor input data from files.

Replace the Python implementation for reading tensor input data from files with
create_sparse_tensor that uses sparse_tensor.new.

The MLIR TNS format has two extra meta data lines. Add the extra meta data to a
test data file.

Implement TACO tensor methods evaluate and unpack.

Add unit tests.

Reviewed By: aartbik

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




More information about the All-commits mailing list