[all-commits] [llvm/llvm-project] 90f22a: [mlir][sparse][taco] Add support for scalar tensors.

[Bixia Zheng via All-commits]

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

  Changed paths:
    A 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] Add support for scalar tensors.

This change allows the use of scalar tensors with index 0 in tensor index
expressions. In this case, the scalar value is broadcast to match the
dimensions of other tensors in the same expression.

Using scalar tensors as a destination in tensor index expressions is not
supported in the PyTACO DSL.

Add a PyTACO test to show the use of scalar tensors.

Reviewed By: aartbik

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