13 Commits

Author	SHA1	Message	Date
Awni Hannun	631dfbe673	fix scatter index bug (#1122)	2024-05-14 15:04:58 -07:00
Awni Hannun	09f1777896	fix slice update indexing (#1053)	2024-04-29 12:17:40 -07:00
Jacket	490c0c4fdc	[Fix] expand axes for dimension with integer indices in mlx_slice_update (#1035) ... * Not sure if this is correct * Format * Edit tests * Add negative test * Format * add one more test --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-04-29 07:57:28 -07:00
Jagrit Digani	639e06e1f3	Indexing bug fix (#947) ... * Fix axes accounting * Add tests	2024-04-01 12:18:50 -07:00
Abdussamet Türker	5611e1a95e	Fix unsqueeze with None (#899) ... * Fix unsqueeze with None * Clean unnecessary files	2024-03-26 13:59:44 -07:00
Jagrit Digani	8e5a5a1ccd	Set item bug fix (#879) ... * set item shaping bug fix * Add extra tests	2024-03-22 12:11:17 -07:00
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
Awni Hannun	9a8ee00246	Switch to nanobind (#839) ... * mostly builds * most tests pass * fix circle build * add back buffer protocol * includes * fix for py38 * limit to cpu device * include * fix stubs * move signatures for docs * stubgen + docs fix * doc for compiled function, comments	2024-03-18 20:12:25 -07:00
Angelos Katharopoulos	961435a243	Scatter vjp (#394) ... * Add a first scatter vjp * Implement the scatter_add vjp * Add array.at to implement user friendly scatters	2024-01-09 13:36:51 -08:00
Awni Hannun	f40d17047d	Indexing bug (#233) ... * fix * test	2023-12-20 10:44:01 -08:00
Awni Hannun	104c34f906	setite negative indexing bug (#189)	2023-12-16 06:44:47 -08:00
Awni Hannun	46a39e5b1f	copyright + ack	2023-11-30 11:12:53 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00