transformers
Enable kernels-community/metal-flash-sdpa on MPS
#45974
Merged

Enable kernels-community/metal-flash-sdpa on MPS #45974

ArthurZucker merged 6 commits into main from enable-metal-flash-sdpa-mps
ArthurZucker
ArthurZucker Enable kernels-community/metal-flash-sdpa on MPS via generate_batch
7d9a893f
HuggingFaceDocBuilderDev
ArthurZucker ArthurZucker requested a review from remi-or remi-or 49 days ago
ArthurZucker serve: auto-select metal-flash-sdpa attention on MPS
36ce80f2
ArthurZucker Revert cu_seqlens synthesis in _flash_attention_forward
f0e30ea1
ArthurZucker serve: pin metal-flash-sdpa to the PR revision until it lands on main
49a0b039
remi-or
remi-or approved these changes on 2026-05-14
ArthurZucker fix
5f638956
ArthurZucker Merge branch 'main' into enable-metal-flash-sdpa-mps
8a52da16
ArthurZucker ArthurZucker enabled auto-merge 35 days ago
github-actions
disabled auto-merge 35 days ago
Manually disabled by user
ArthurZucker ArthurZucker merged bdca59aa into main 35 days ago
ArthurZucker ArthurZucker deleted the enable-metal-flash-sdpa-mps branch 35 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone