[all-commits] [llvm/llvm-project] c9ff83: [mlir][Linalg] Fix linalg.generic iteration domain...

[Kunwar Grover via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c9ff8399647cd15cdb9f8853b45854920de17162
      https://github.com/llvm/llvm-project/commit/c9ff8399647cd15cdb9f8853b45854920de17162
  Author: Kunwar Grover <groverkss at gmail.com>
  Date:   2025-02-20 (Thu, 20 Feb 2025)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
    M mlir/test/Dialect/Linalg/fuse-with-reshape-by-collapsing.mlir
    M mlir/test/Dialect/Linalg/fusion-push-reshape.mlir

  Log Message:
  -----------
  [mlir][Linalg] Fix linalg.generic iteration domain collapse for dynamic dims (#118208)

This pr fixes how iteration domain of linalg.generic is collapsed when
fusing with tensor.expand_shape. Previously, the output_shape for
tensor.expand shape was infered, which doesn't always work except some
special cases.

This patch makes the logic explicitly set the bounds of the new
collapsed iteration domain, because we already know them.

---------

Co-authored-by: Jakub Kuderski <kubakuderski at gmail.com>



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications