7 Commits

Author	SHA1	Message	Date
Awni Hannun	c1eb9d05d9	non-symmetric eig and eigh (#2188)	2025-05-15 13:01:44 -07:00
Awni Hannun	f1606486d2	Generalize gpu backend (#2138) ... * generalize gpu backend * fix no_gpu build * fix no_gpu build * generalize gpu backend	2025-04-30 09:08:17 -07:00
Awni Hannun	f2c85308c1	add a half simd gemm fallback (#2046) ... * add a half simd gemm fallback * nit	2025-04-07 09:31:29 -07:00
Awni Hannun	de5f38fd48	Custom logsumexp (#2028) ... * initial custom logsumexp * more tests * comments + fix	2025-03-31 07:36:55 -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
Abe Leininger	a5ededf1c3	CPU LU factorization and linear solvers (#1451) ... * linalg solve backend * nits * more nits + fix * luf primitive and lu, solve, and solve_triangular backends * changes / nits --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-02-10 12:32:24 -08:00
Awni Hannun	1156c84e86	Refactor common into cpu specific and truly common (#1817) ... * refactor * fix extension example * fix no-cpu	2025-02-03 15:58:02 -08:00