[all-commits] [llvm/llvm-project] d7904a: [MLIR] Fold outer dims permutation to pack when pr...

[lorenzo chelini via All-commits]

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

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/DataLayoutPropagation.cpp
    M mlir/test/Dialect/Linalg/data-layout-propagation.mlir

  Log Message:
  -----------
  [MLIR] Fold outer dims permutation to pack when propagating

Instead of folding the transpose into the linalg.generic keep the
transposition in the packing operation, effectively making the
linalg.generic transparent to the propagation. Additionally, if the init
operand of the generic has users pack the init and pass it as the
operand to the generic.

Reviewed By: hanchung

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