20 Commits

Author	SHA1	Message	Date
Awni Hannun	4e1e9520e1	Flatten and unflatten (#1692) ... * flatten and unflatten * fix grad * fix shape infer * use squeeze + unsqueeze in get_item	2024-12-11 21:51:37 -08:00
Cheng	0bf19037ca	Remove "using namespace mlx::core" in python/src (#1689)	2024-12-11 15:45:39 -08:00
Awni Hannun	f76a49e555	ExpandDims primitive (#1687) ... * add squeeze primitive * simplify squeeze, use in gather * fix * fix * fix * fix * fix no cpu * use squeeze in matmul and friends * expand dims primitive * comment	2024-12-10 16:39:07 -08:00
Cheng	87d7a2520e	Use Py_ssize_t in python bindings (#1678) ... * Use Py_ssize_t in python bindings * Args passed to std::max must be same type	2024-12-09 12:59:19 -08:00
Awni Hannun	09bc32f62f	No extra reshape (#1557) ... * no extra reshape * lint	2024-11-02 19:07:20 -07:00
Awni Hannun	40b6d67333	Fixes for large arrays with a few ops (#1299) ... * fixes for large arrays with a few ops * fix bug * fix all of copy	2024-07-30 17:18:39 -07:00
Awni Hannun	eb8321d863	list based indexing (#1150)	2024-05-22 15:52:05 -07:00
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