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

Quinn Dawkins via All-commits all-commits at lists.llvm.org
Tue May 2 09:29:37 PDT 2023

  Branch: refs/heads/main
  Home:   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

More information about the All-commits mailing list