zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
19fb69e2ed
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