mlx/python
Jagrit Digani a5681ebc52
Update set item (#861)
* Update mlx_set_item to handle regular slices without expanding

* Refactor ellipsis handling

* Route mlx_set_item to slice_update where possible

* Update mlx_scatter_args_slice

* Don't route to gather if no array indices
2024-03-21 02:48:13 -07:00
..
mlx No reshape rope (#838) 2024-03-18 17:03:07 -07:00
src Update set item (#861) 2024-03-21 02:48:13 -07:00
tests Check edge case handling in row reduce med kernel (#858) 2024-03-20 11:37:58 -07:00