Code

[WIP] Vae fusion improvements.#19679

Code

Draft

MaheshRavishankar wants to merge 2 commits intoiree-org:mainiree-org/iree:mainfrom MaheshRavishankar:vae_fusionMaheshRavishankar/iree:vae_fusionCopy head branch name to clipboard

+54-4

Commits

Commits on Jan 10, 2025

• Reshape propagation to enable broadcast(transpose) -> attention(q, kt, vt) fusion.

  Show description for 640a83f
  MaheshRavishankar

  committed
  640a83f

  Copy full SHA for 640a83f

• Propagate expand_shape past collapse_shape

  Show description for 8dd1bf9
  

  MaheshRavishankar

  authored and

  MaheshRavishankar

  committed
  8dd1bf9

  Copy full SHA for 8dd1bf9