Code

Legacy flash remat fix#943

Code

Merged

hanzhi713 merged 2 commits intomainapple/axlearn:mainfrom legacy-flash-remat-fixapple/axlearn:legacy-flash-remat-fixCopy head branch name to clipboardJan 23, 2025

+25-19

Commits

Commits on Jan 22, 2025

• Fix the same problem for legacy tpu attn

  hanzhi713

  committed
  eac5205

  Copy full SHA for eac5205

Commits on Jan 23, 2025

• Fix

  hanzhi713

  committed
  3b701fb

  Copy full SHA for 3b701fb