[all-commits] [llvm/llvm-project] 358072: [mlir][sparse][taco] Support the use of index valu...

[Bixia Zheng via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 3580721a59d9a6db365978d176044dc21a7eb3fa
      https://github.com/llvm/llvm-project/commit/3580721a59d9a6db365978d176044dc21a7eb3fa
  Author: Bixia Zheng <bixia at google.com>
  Date:   2022-03-15 (Tue, 15 Mar 2022)

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

  Log Message:
  -----------
  [mlir][sparse][taco] Support the use of index values in tensor expressions.

PyTACO DSL doesn't support the use of index values as in A[i] = B[i]+ i.
We extend the DSL to support such a use in MLIR-PyTACO.

Remove an obsolete unit test. Add unit tests and PyTACO tests.

Reviewed By: aartbik

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