[all-commits] [llvm/llvm-project] 2f49e6: Support sparse tensor output.

Bixia Zheng via All-commits all-commits at lists.llvm.org
Mon Dec 13 12:06:47 PST 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2f49e6b0dbf7bdcda02771f1f7f845f948378999
      https://github.com/llvm/llvm-project/commit/2f49e6b0dbf7bdcda02771f1f7f845f948378999
  Author: Bixia Zheng <bixia at google.com>
  Date:   2021-12-13 (Mon, 13 Dec 2021)

  Changed paths:
    M mlir/lib/ExecutionEngine/SparseTensorUtils.cpp
    A mlir/test/Integration/Dialect/SparseTensor/python/test_elementwise_add_sparse_output.py
    A mlir/test/Integration/Dialect/SparseTensor/python/tools/lit.local.cfg
    A mlir/test/Integration/Dialect/SparseTensor/python/tools/np_to_sparse_tensor.py

  Log Message:
  -----------
  Support sparse tensor output.

Add convertFromMLIRSparseTensor to the supporting C shared library to convert
SparseTensorStorage to COO-flavor format.

Add Python routine sparse_tensor_to_coo_tensor to convert sparse tensor storage
pointer to numpy values for COO-flavor format tensor.

Add a Python test for sparse tensor output.

Reviewed By: aartbik

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




More information about the All-commits mailing list