Enabled running Pallas Flash Attention on CPU.#922

Merged

ds-hwang merged 1 commit intoapple:mainfrom ds-hwang:flsh_cpu

Jan 23, 2025

+82-40

Commits on Jan 22, 2025

Enabled running Pallas Flash Attention on CPU.
ds-hwang
committed