zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-26 07:48:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
19fb69e2ed3344554b50d9435c95083906703e69
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