[all-commits] [llvm/llvm-project] a44b78: [MLIR][linalg] Fix unpack rewriter for dynamic sha...

qcolombet via All-commits all-commits at lists.llvm.org
Fri Sep 22 03:24:00 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: a44b787e06c203eecb82920c331ef0de751501eb
      https://github.com/llvm/llvm-project/commit/a44b787e06c203eecb82920c331ef0de751501eb
  Author: qcolombet <quentin.colombet at gmail.com>
  Date:   2023-09-22 (Fri, 22 Sep 2023)

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

  Log Message:
  -----------
  [MLIR][linalg] Fix unpack rewriter for dynamic shapes (#67096)

Prior to this patch, `GeneralizeOuterUnitDimsUnPackOpPattern` would
assert that we cannot create a `tensor.empty` operation with dynamic
shapes.

The problem stems from the fact that we were not using the right builder
for the `tensor.empty` operation. Indeed, each dynamic dim needs to be
specified by an input variable.

Simply provide the dynamic dimensions to the `tensor.empty` builder to
fix that.




More information about the All-commits mailing list