[all-commits] [llvm/llvm-project] 6eee66: [mlir][linalg] Add a new pattern to handle folding...

Han-Chung Wang via All-commits all-commits at lists.llvm.org
Wed Nov 23 10:47:32 PST 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 6eee66d12ab33f35a37a1514342b51ae93d175e8
      https://github.com/llvm/llvm-project/commit/6eee66d12ab33f35a37a1514342b51ae93d175e8
  Author: Hanhan Wang <hanchung at google.com>
  Date:   2022-11-23 (Wed, 23 Nov 2022)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp
    M mlir/test/Dialect/Linalg/drop-unit-extent-dims.mlir
    M utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

  Log Message:
  -----------
  [mlir][linalg] Add a new pattern to handle folding unit reduction dims.

The output operands will be added to input operands if the generic op (on tensors)
becomes an elementwise operation. The outputs of the generic op is still the same.
They will be cleaned up by ReplaceWithEmptyTensorIfUnused pattern.

Reviewed By: mravishankar

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




More information about the All-commits mailing list