1488 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Cheng	b704e9e77a	[CUDA] Check CUDA error in synchronize (#2757)	2025-11-14 07:10:23 +09:00
Awni Hannun	66519fb348	fix slice (#2758)	2025-11-13 11:30:02 -08:00
Awni Hannun	8973550ff3	export custom kernel (#2756)	2025-11-13 11:29:50 -08:00
Mike Drob	3f866be665	minor debugging for publishing (#2739) ... * minor debugging for publishing * fix logic	2025-11-12 06:33:39 -08:00
Awni Hannun	23f81ed1c1	Linux on arm (#2751) ... * try linux on arm * ssh * fix	2025-11-11 11:41:14 -08:00
wrmsr	3fe2250c00	Fix irregular_strides benchmark shape type (#2754)	2025-11-11 11:40:22 -08:00
Awni Hannun	047114b988	remove circle (#2753)	2025-11-11 11:39:47 -08:00
wrmsr	9320eb89a8	Fix dequantize python sig (dtype default) (#2752)	2025-11-11 09:55:24 -08:00
Awni Hannun	75819d70ea	patch bump (#2750)	2025-11-11 08:49:14 -08:00
Awni Hannun	60d80a3728	fix release builds (#2746) v0.29.4	2025-11-11 07:44:30 -08:00
Pedro Cuenca	eba6a9d163	Compatibility with pip-installed openmpi (#2741)	2025-11-07 16:58:31 -08:00
CCYeh	be9e2aebd6	Shapeless support for zeros/ones_like (#2726) ... * shapeless support for zeros/ones_like * Improvements * fix access after moved	2025-11-06 19:12:20 -08:00
Awni Hannun	df58b4133a	[CUDA] Reduce use of managed memory (#2725) ... * Use async cuda malloc managed with cuda 13 * add pool threshold * refactor for regular cuda malloc * load eval gpu for cuda * remove use of cuda pool, use cuda free async * fix * fix * fix * fix * fix + comment	2025-11-05 16:05:23 -08:00
Anastasiia Filippova	27778156dc	Nccl reduce scatter, all gather (#2727) ... * Added reduce scatter and all gather for nccl * fix unused import, delete unused file * small fix * deleted useless condition * fixed comments * fix bug in eval_gpu, renamed to sum_scatter, fix docs * final fix docs * remove and * Update mlx/distributed/mpi/mpi.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * fix broken set input output * fixes set output * typo * fix typo * no cpu, no gpu for reduce scatter --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com>	2025-11-05 08:21:11 -08:00
Mike Drob	761f901a41	fix property name (#2736)	2025-11-05 06:31:56 -06:00
Angelos Katharopoulos	6ece97f69b	Make cpu binary_op easily accessible (#2733)	2025-11-05 01:08:41 -08:00
Awni Hannun	d3bc6a9bff	don't test when doing release (#2734)	2025-11-04 15:54:23 -08:00
Awni Hannun	26ceb507eb	only build for macos 14 and up (#2731) ... * only build for macos 14 and up * bump metal cpp	2025-11-04 09:44:15 -08:00
Mike Drob	910b3e3299	skip self-hosted runners on forks (#2730)	2025-11-03 16:22:13 -06:00
Harsh Sutaria	50fa315d18	Fix addmm with empty matrices and beta != 1.0 (#2715)	2025-11-03 14:16:15 -08:00
AN Long	1ff2b713b6	Check isnan in maximum / minimum with CPU backend (#2652) ... * Check isnan in maximum / minimum with CPU backend * Add tests * fix --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-11-03 08:51:14 -08:00
Mike Drob	50514a6146	Set up publishing to PyPI and Test-PyPI (#2721)	2025-11-03 07:20:11 -08:00
Awni Hannun	93d76b0f30	Fix compile multi capture (#2678) ... * fix compile when compiling multiple lambdas with the same capture * add test	2025-11-03 06:33:43 -08:00
David Koski	78678de0cd	add null check -- the bundleIdentifier is optional (#2709) ... * add null check -- the bundleIdentifier is optional * use variable	2025-11-03 06:33:21 -08:00
Melissa Kilby	ed9c6b1117	update: add linux fedora container CI - CPP build test only (#2722) ... * update: add linux_fedora_build_cpp CI - CPP build test only - x86-64 Signed-off-by: Melissa Kilby <mkilby@apple.com> * update: add linux_fedora_build_cpp_aarch64 CI - CPP build test only - arm64 Co-authored-by: Mike Drob <mdrob@apple.com> Signed-off-by: Melissa Kilby <mkilby@apple.com> * update: convert linux_fedora_build_cpp to matrix.arch loop Co-authored-by: Mike Drob <mdrob@apple.com> Signed-off-by: Melissa Kilby <mkilby@apple.com> --------- Signed-off-by: Melissa Kilby <mkilby@apple.com> Co-authored-by: Mike Drob <mdrob@apple.com>	2025-11-03 06:33:00 -08:00
Awni Hannun	39b04ce638	use faster dequant for fp4 qmv (#2720)	2025-10-31 11:49:59 -07:00
Mike Drob	d9e6349657	fix docs path (#2719)	2025-10-30 19:12:49 -05:00
Angelos Katharopoulos	b901a9f311	Fix the order of hosts in the ring (#2718)	2025-10-30 15:02:39 -07:00
Awni Hannun	68c5fa1c95	fix memory count bug (#2717)	2025-10-30 14:27:15 -07:00
Christopher Webb	793a31eeb6	Fix missing domain_uuid_key in thunderbolt ring setup (#2682)	2025-10-30 13:17:20 -07:00
Mike Drob	74c1ed25bb	Migrate CircleCI to GitHub Actions (#2716) ... Co-authored-by: Joseph Heck <j_heck@apple.com>	2025-10-30 12:26:55 -05:00
Awni Hannun	ec72b44417	Add quantize/dequantize for mxfp8 and nvfp4 (#2688) ... * Add quantize/dequantize slow path for mxfp8 and nvfp4 * fast cuda kernel for mx/nv quantization * fallback for cuda < 12.8 (#2697) * format (#2700) * fix (#2701) * metal kernels * docs * fix jit * add default bits and group sizes * improve quant docs * fix output type of mxfp4 matmuls	2025-10-28 16:23:12 -07:00
Melissa Kilby	460691a0e8	fix: linux-{fedora}x86_64-build (#2707) ... Signed-off-by: Melissa Kilby <mkilby@apple.com>	2025-10-27 16:36:08 -07:00
Awni Hannun	969924cc69	Fp8 conversion (#2686) ... * add fp8 e4m3 converters * add cuda * default saturate to min/max * fix for older OS * fix no gpu/cpu * fix saturate * fix compile	2025-10-27 16:35:50 -07:00
Awni Hannun	d1e06117e8	bump python (#2694)	2025-10-27 11:34:31 -07:00
Awni Hannun	539d8322d1	add median op (#2705)	2025-10-27 11:33:42 -07:00
Awni Hannun	c4767d110f	fix addmm cpu (#2699)	2025-10-27 11:33:32 -07:00
David Koski	895217f25b	optionally load metallib from framework (#2702) ... * optionally load metallib from framework * pre-commit * adjust logic	2025-10-27 07:52:03 -07:00
Manuel Villanueva	0cfeeb60ca	Einsum error msg improvement (#2690) ... * Improved error message for Einsum * Modifications via pre-commit * format * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-10-27 06:31:47 -07:00
Ronan Collobert	8f8af61a37	fix warnings showing up with -Wall (#2692)	2025-10-24 11:43:35 -07:00
Manuel Villanueva	233384161e	Improved mx.split() docs (#2689) ... * Improved mx.split() documentation * Fix typo in docstring for array split function * add example --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-10-24 09:48:41 -07:00
Awni Hannun	5bcf3a6794	format	2025-10-22 16:08:47 -07:00
wickedcoder	7707196297	Merge commit from fork ... * add length validation to the header * fix accessing out of bound index with .at()	2025-10-22 15:31:25 -07:00
wickedcoder	7e3471c987	Merge commit from fork ... * add tensor->weights_data validation * add null pointer check for tensor	2025-10-22 15:31:03 -07:00
Awni Hannun	9f0ba3ddf1	patch bump (#2680)	2025-10-17 12:12:07 -07:00
Awni Hannun	4bce5f9b2d	suppress gcc 10.1 warnings (#2679) ... * suppress gcc 10.1 warnings * suppress gcc 10.1 warnings v0.29.3	2025-10-17 12:09:21 -07:00
Anastasiia Filippova	e9eab527eb	Nccl timeout (#2673) ... * print the error & delete nccl group * timeout for nccl binding * typo * revert error * fixed a typo	2025-10-14 12:29:54 -07:00
Awni Hannun	36ca62dba8	remove unused unary file (#2672)	2025-10-13 19:36:26 -07:00
Manuel Villanueva	9cbb1b0148	Modified sort behavior when running CPU or Metal to match NumPy/JAX (#2667) ... * Modified sort behavior when running CPU or Metal to match NumPy/JAX sorting behavior. * Modified sort behavior when running CPU or Metal to match NumPy/JAX * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-10-13 14:36:45 -07:00
Fabrizio Milo	9bfc476d72	Normalize README bullet formatting (#2671)	2025-10-13 12:13:30 -07:00