Skip to content

[Fusion] Add HorizontalFusion pass. - #452

Open
charithaintc wants to merge 7 commits into
intel:mainfrom
charithaintc:horizontal_fusion_impl
Open

charithaintc wants to merge 7 commits into
intel:mainfrom
charithaintc:horizontal_fusion_impl

Conversation

@charithaintc

Copy link
Copy Markdown
Collaborator

This PR adds the horizontal fusion capability to matmul and grouped matmul ops. Horizontal fusion is decoupled from tiling and fusion passes, instead it happens at Linalg op level. Eg: Two linalg.matmul ops sharing an input will be fused to a linalg.generic with two outputs.

Here is a summary of main changes.

  1. Update decomposition pass to decompose fused matmul generics and fused grouped matmul generics to individual matmuls.
  2. Update grouped matmul op defn to support fused form (multi-input multi-output groped matmul)
  3. Updated tiling to support fused generic form.
  4. Add horizontal fusion pass that runs before tiling and fusion.

Assisted by: Codex, Claude

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant