Preserve SDPA for CoreML#5258

Closed

YifanShenSZ wants to merge 3 commits intopytorch:mainfrom

YifanShenSZ:coreml-sdpa

Commits on Sep 11, 2024

add coreml custom sdpa to avoid decomposition
yifan_shen3
committed

Commits on Sep 12, 2024

also use torch.index_put_ in KV cache, to cleanly map to coreml iOS18.slice_update
yifan_shen3
committed
fix lint
yifan_shen3
committed