zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-24 12:18:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
343 Commits 45 Branches 73 Tags
fe1dabf272369456d95729733ca916a297ecb7bf
Commit Graph

2 Commits

Author SHA1 Message Date
Vijay Krish
06072601ce Scatter optimization : Eliminate 64b integer divide. (#662) 
Launch 2D grid to eliminate divide and mod in device code,
since 64b integer division is very expensive.

Github Issue #506

Co-authored-by: Vijay Krishnamoorthy <vijay_krish@apple.com>
 2024-02-10 08:49:51 -08:00
Awni Hannun
e319383ef9 Faster gather (#626) 
* faster gather

* update copyright
 2024-02-04 17:25:44 -08:00