Rewrite prefix key-padding masks as k/v truncation to keep SDPA on the flash path (PyTorch) - #80
Open
arnaudlvq wants to merge 1 commit into
Open
Google CLA / cla/google
succeeded
Jul 21, 2026 in 20s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
The following contributors were found for this pull request:
✅ 4078baa Committer: @arnaudlvq <arna*******que@epfl.ch>, <ar*****vq@gmail.com>
✅ 4078baa Author: <arn*****vq@gmail.com>
(Only the first commit for a unique contributor is listed.)
Loading