Awni Hannun 005e7efa64 fix mask in sdpa (#1980) ...

* fix mask in sdpa

* fix attention mask

* Re-enable routing for array mask

---------

Co-authored-by: Jagrit Digani <digani@apple.com>

2025-03-20 14:53:12 -07:00

18 KiB

Raw Blame History

View Raw