[all-commits] [llvm/llvm-project] 6f0719: [mlir][sparse][taco] Support reduction to scalar t...

Bixia Zheng via All-commits all-commits at lists.llvm.org
Fri Feb 25 14:17:58 PST 2022


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

  Changed paths:
    M mlir/test/Integration/Dialect/SparseTensor/taco/test_scalar_tensor_algebra.py
    M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py

  Log Message:
  -----------
  [mlir][sparse][taco] Support reduction to scalar tensors.

The PyTACO DSL doesn't support reduction to scalars. This change
enhances the MLIR-PyTACO implementation to support reduction to scalars.

Extend an existing test to show the syntax of reduction to scalars and
two methods to retrieve the scalar values.

Reviewed By: aartbik

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




More information about the All-commits mailing list