commit 9c951cac8eddc9828adadd7d7bd4207df5f145b4
author MaheshRavishankar <1663364+MaheshRavishankar@users.noreply.github.com> Tue Aug 13 09:05:10 2024 -0700
committer GitHub <noreply@github.com> Tue Aug 13 09:05:10 2024 -0700
tree b66758ae6c2ac53d6342a83a3612c27a01ccd305
parent 7812c776d5d57b13d80b6ae27f2dc86c73fddbcf

[Flow] Generalize horizontal contraction fusion to cover more cases.  (#17880)

Current implementation of horizontal fusion worked only for
contraction (-> truncf)? cases. To generalize it to more cases the
following changes were needed

1) Instead of looking for a `linalg.generic` with a single
   `arith.truncf`, use `isBitTruncate` utility method. To enable this
   the pass is moved to `Flow`.

2) Use `Operation::Equivalence` to check that the contraction
   operations are "similar" and any subsequent truncation operations
   are "similar" too.

3) Instead of trying to find an insertion point based on existing
   dominance relationship between operations,
   - Always insert the horizontally fused contraction before the first
     contraction
   - Always insert the horizontally fused truncation operation before
     the first truncation operation

4) Instead of generating the fills/empty along with horizontally fused
   operations, use separate patterns to fold concats of fills and
   concats of emptys into fill and empty, respectively.

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>

---------

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>