[all-commits] [llvm/llvm-project] ae7ee6: [mlir][taco] Add a utility to create an MLIR spars...

Bixia Zheng via All-commits all-commits at lists.llvm.org
Tue Feb 1 15:44:06 PST 2022


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

  Changed paths:
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py
    A mlir/test/Integration/Dialect/SparseTensor/taco/unit_test_tensor_utils.py

  Log Message:
  -----------
  [mlir][taco] Add a utility to create an MLIR sparse tensor from a file.

Move the functions that retrieve the supporting C library, compile an MLIR
module and build a JIT execution engine to mlir_pytaco_utils.

Add a function to create an MLIR sparse tensor from a file and return a pointer
to the MLIR sparse tensor as well as the shape of the sparse tensor.

Add unit tests.

Reviewed By: aartbik

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




More information about the All-commits mailing list