1474 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
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
Awni Hannun	25e2356316	speed up scalars (#2669)	2025-10-13 12:10:15 -07:00
Awni Hannun	226a1d24e0	Debug cuda conv (#2662) ... * use t4 * use t4	2025-10-10 16:12:47 -07:00
Awni Hannun	630350ad3e	Precise sigmoid (#2659) ... * bump patch * Sigmoid matches PyTorch and is more precise on tails	2025-10-10 10:05:23 -07:00
Awni Hannun	380aeb58ae	enable admm low-precision cpu (#2661)	2025-10-10 09:50:54 -07:00
Awni Hannun	f37389d100	bump patch (#2658)	2025-10-10 08:36:41 -07:00
Awni Hannun	e89e8b4272	Export with callback (#2612) ... * export with callback * export with callback * Add types, fix kwarg ordering bug + test * cleanup, test, fix * typos	2025-10-08 19:24:33 -07:00
AN Long	85a8824a8c	Fix cumulative operations when axis=None (#2653)	2025-10-08 15:25:38 -07:00
Awni Hannun	f5d4397e5c	Fix fast synch when fence is waited before a command buffer is created (#2657)	2025-10-08 11:23:46 -07:00
Awni Hannun	343e33b6d5	fix all_gather vjp (#2654)	2025-10-07 06:05:23 -07:00
Angelos Katharopoulos	0073096dd1	Split name into directories for cuda jit (#2656)	2025-10-07 01:52:58 -07:00
Angelos Katharopoulos	e3d004fed9	Fix and refactor row-reduce (#2650)	2025-10-07 01:51:08 -07:00
Awni Hannun	a393435d28	Speed up compile for node with many parents (#2649)	2025-10-03 19:30:36 -07:00
Awni Hannun	a7a94b29d7	Fix compile when outputs change (#2648)	2025-10-03 08:40:57 -07:00
Daniel Yeh	22a5da76c8	Faster complex matmul (#2571)	2025-10-02 23:33:15 -07:00