[all-commits] [llvm/llvm-project] 53cc3a: [mlir][sparse] index support in sparse compiler co...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 53cc3a06378229f5b4713f0db39135e846609d0a
      https://github.com/llvm/llvm-project/commit/53cc3a06378229f5b4713f0db39135e846609d0a
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-03-08 (Tue, 08 Mar 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/Utils/Merger.h
    M mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
    M mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
    A mlir/test/Dialect/SparseTensor/sparse_index.mlir
    A mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_index.mlir

  Log Message:
  -----------
  [mlir][sparse] index support in sparse compiler codegen

This revision adds support for the linalg.index to the sparse compiler
pipeline. In essence, this adds the ability to refer to indices in
the tensor index expression, as illustrated below:

 Y[i, j, k, l, m] = T[i, j, k, l, m]  * i * j

Reviewed By: bixia

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