Code

[DRAFT] Generalize MHA pattern#2092

Code

Open

gramalingam wants to merge 5 commits intomainmicrosoft/onnxscript:mainfrom rama/MHAmicrosoft/onnxscript:rama/MHACopy head branch name to clipboard

+173-107

Commits

Commits on Mar 6, 2025

• Part 1 of MHA refactoring

  gramalingam

  committed
  fb04ef0

  Copy full SHA for fb04ef0

• Partial changes for 3D rotary embedding

  gramalingam

  committed
  1f5c569

  Copy full SHA for 1f5c569

Commits on Mar 7, 2025

• Updates to MHA

  gramalingam

  committed
  075baf4

  Copy full SHA for 075baf4

• Fix ort version dependence and run lint

  gramalingam

  committed
  a20752f

  Copy full SHA for a20752f

• Replace old MHA by new

  gramalingam

  committed
  64fad95

  Copy full SHA for 64fad95