10 Commits

Author	SHA1	Message	Date
Awni Hannun	0c5eea226b	Reduce specializations (#1607) ... * start of reduce specializations * fix all reduce * fix many dims * fix * non-jit tests clear * cleanup instantiations * cpu merges * change dim specializations * optimize * fix jit * fix jit * use higher precision for integer sum+prod * fixes	2024-11-21 19:53:00 -08:00
Angelos Katharopoulos	969337345f	Fix reduce edge case (#1389)	2024-09-01 21:37:51 -07:00
Angelos Katharopoulos	8081df79be	Fix boolean all reduce bug (#1355)	2024-08-24 10:09:32 -07:00
Angelos Katharopoulos	b57a52813b	Further reduction tuning (#1349) ... * More reduction tuning * Forgotten pdb * Small column long row specialization	2024-08-23 10:35:25 -07:00
Awni Hannun	98b6ce3460	Refactor reductions and fix scatter atomics for large sizes (#1300) ... Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-08-22 16:03:31 -07:00
Vijay Krish	fcc5ac1c64	Add GPU support for uint64/int64 reductions (#569)	2024-01-31 11:18:04 -08:00
Awni Hannun	b9e415d19c	bump pre commit and fix format (#373)	2024-01-04 16:28:52 -08:00
Zach Schillaci	5b9be57ac3	Add isort pre-commit and run (#68)	2023-12-08 11:31: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