[all-commits] [llvm/llvm-project] 009c05: [mlir][linalg] Allow outer dims perm and untiled d...

[Quinn Dawkins via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 009c053e3f822d0df556c6b39f632e31594373de
      https://github.com/llvm/llvm-project/commit/009c053e3f822d0df556c6b39f632e31594373de
  Author: Quinn Dawkins <quinn at nod-labs.com>
  Date:   2023-05-02 (Tue, 02 May 2023)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
    M mlir/test/Dialect/Linalg/generalize-tensor-pack.mlir
    M mlir/test/Dialect/Linalg/generalize-tensor-unpack.mlir

  Log Message:
  -----------
  [mlir][linalg] Allow outer dims perm and untiled dims in pack/unpack generalization

Extends the pack/unpack generalization patterns to work for any packing
op with only full tiles. This produces a combination of rank-reduced
insert/extract slice ops paired with a transpose on the reduced shape,
similar to what the pattern currently produces for fully tiled
pack/unpacks. Note that only the outer dims are rank-reduced in this
pattern, leaving the shape of the inner tile intact.

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