13 Commits

Author	SHA1	Message	Date
Angelos Katharopoulos	99eefd2ec0	Gather mm new kernel and small refactoring (#2040)	2025-04-14 16:37:36 -07:00
Awni Hannun	7b7e2352cd	fix malloc or wait deadlock (#1976)	2025-03-20 16:48:43 -07:00
Awni Hannun	c4230747a1	redesign for faster cpu/gpu synch (#1869) ... * redesign for faster cpu/gpu synch * load + more async CPU * use command encoder API and move more ops to use it * make fence back-end generic + CPU only fence * faster build * fix async eval * fixes + handle temporaries * fix / improve cpu conv * remove unused status, fix siblings * fix extensions * fix * fix no cpu build * format * comments * fix perf regression, remove unecessary abort * fix events, task limit cpu * fix waiting * fix donation / temporaries in normalization	2025-03-06 19:23:38 -08:00
Awni Hannun	1c0c118f7c	Fp64 on the CPU (#1843) ... * add fp64 data type * clean build * update docs * fix bug	2025-02-07 15:52:22 -08:00
Awni Hannun	1ccaf80575	Dynamic broadcasting for shapeless compile/export (#1722) ... * working towards dynamic broadcast * shapeless broadcast * fix build + nits * use broadcast arrays in quantize matmul * some cleanup / consistency * mend * some comments * add vjp, jvp for broadcast axes	2025-01-09 11:04:24 -08:00
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
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
Awni Hannun	40c62c1321	Use int64 stride everywhere (#1671) ... * use int64 stride everywhere * fix ext * fix ext * more shape + cleanup * one more * few more	2024-12-09 11:09:02 -08:00
Awni Hannun	9bd03dd9b4	More buffer donation with no-ops (#1591) ... * more donation * fix test * fix build	2024-11-18 08:35:41 -08:00
Awni Hannun	5b6f38df2b	Faster cpu ops (#1434) ... * faster binary and cleaner copy * use recursive template for other ops * more cleanup * fix from cleanup * more clean * fix binary * use contiguous iterator * add 3d * nits * fix * fix? * fix * fix rebase	2024-09-26 09:19:13 -07:00
Angelos Katharopoulos	5c1fa64fb0	Custom transforms (#1246)	2024-07-10 18:00:01 -07:00
Alex Barron	375a8bbdcc	Add some internal GPU apis (#1177) ... * Add unary/binary/ternay/slice/concat internal GPU ops * add pad internal op * formatting + no_cpu fix	2024-06-04 09:24:26 -07:00
Awni Hannun	7178ac0111	No CPU option for binary minimization (#1105) ... * no cpu build option * docs * fix	2024-05-13 16:08:11 -07:00