[all-commits] [llvm/llvm-project] e383ea: [mlir][sparse] parameterize MTTKRP kernel

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: e383eaa647da3a3f45313d22d83b6d6b5cdc9fc1
      https://github.com/llvm/llvm-project/commit/e383eaa647da3a3f45313d22d83b6d6b5cdc9fc1
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-01-14 (Fri, 14 Jan 2022)

  Changed paths:
    M mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_mttkrp.mlir

  Log Message:
  -----------
  [mlir][sparse] parameterize MTTKRP kernel

Rather than hardcoding all constants, we now use the input tensor to drive the
code setup. Of course, we still need to hardcode dim-2 of A and the final
verification in CHECK is input dependent, but overall this sets a slightly
better example of tensor setup in general.

Reviewed By: bixia

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