[all-commits] [llvm/llvm-project] d5a9fc: [MLIR] Fix tiling for `tensor.unpack` with outer p...

[lorenzo chelini via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d5a9fc13ef82f7f6c1a350bceef79f7988cdac20
      https://github.com/llvm/llvm-project/commit/d5a9fc13ef82f7f6c1a350bceef79f7988cdac20
  Author: Lorenzo Chelini <l.chelini at icloud.com>
  Date:   2023-01-16 (Mon, 16 Jan 2023)

  Changed paths:
    M mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp
    M mlir/test/Dialect/Tensor/tiling.mlir

  Log Message:
  -----------
  [MLIR] Fix tiling for `tensor.unpack` with outer permutations

An outer dim permutation requires adjusting the offsets and sizes of the
`tensor.extract_slice` operations generated during tiling. Originally
this was done by computing an inverse permutation of the outer
permutation for both `tensor.pack` and `tensor.unpack`. For packing, the
tiling is applied on interchanged dimensions; thus, it is correct to
compute the inverse. For unpacking, on the other hand, tiling involves
the output tensor that does not have interchanged dimensions, and no
inverse is required.

Reviewed By: hanchung

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