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
ebfcb4a14f3845bc100585061e3054a9157fc01b
mlx/python/tests/test_fast_sdpa.py
Awni Hannun e613d0eaf0 SDPA support for small batch (over sequence) queries (#1922) 
* batch query sdpa

* batch sdpa for query
2025-03-04 10:59:04 -08:00

11 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink