Re-apply Fixed issue with dot_product_attention when using TPU. #21254 after addressing cuDNN/FlashAttention API updates #21333
Google CLA / cla/google
succeeded
Jun 10, 2025 in 2s
✅ 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:
✅ a927e7e Author: @pctablet505 <hel*****ul@google.com>, <55033230+pctablet505@users.noreply.github.com>, <pct******05@gmail.com>
✅ de81e5b Co-Author: @JyotinderSingh <33001894+JyotinderSingh@users.noreply.github.com>
(Only the first commit for a unique contributor is listed.)
Loading