zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-11 15:06:42 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
Files
1086dc4db070d4f7d211f2bb760f560b98f969f7
mlx/python
History
Brian Keene 19fb69e2ed Add memory_efficient_threshold kwarg to sdpa kernel (#1319) 
Allows opt-in to memory efficient GPU shader at proscribed sequence
length.  Otherwise, utilizes aggregate MLX primitives for best latency.
2024-08-12 12:57:09 -07:00
..
mlx
Add "edge" mode to mx.pad (#1309)
2024-08-06 11:23:10 -07:00
src
Add memory_efficient_threshold kwarg to sdpa kernel (#1319)
2024-08-12 12:57:09 -07:00
tests
Add memory_efficient_threshold kwarg to sdpa kernel (#1319)
2024-08-12 12:57:09 -07:00