zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
a982077f6e57a6a13f09d54b110f0ffb502e1b7c
mlx/python/tests/test_fast_sdpa.py
Awni Hannun d6977f2a57 Add sdpa with sinks (#2558) 
* add sdpa with sinks

* fix 2 pass

* fix matrix sdpa

* fix perf regression

* add to cuda (#2580)
2025-09-10 14:53:00 -07:00

26 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink