8 Commits

Author	SHA1	Message	Date
Fangjun Kuang	f20e97b092	minor fixes (#1194) ... * minor fixes * fix build errors	2024-06-12 22:06:49 -07:00
Awni Hannun	0189ab6ab6	More jitting (#1132) ... * docs + circle min size build * jit scan, arange, softmax * add sort * jit reductions * remove print * fix deps * clean includes / nits	2024-05-23 16:23:44 -07:00
Awni Hannun	226748b3e7	JIT compile option for binary minimization (#1091) ... * try cpp 20 for compile * unary, binary, ternary in jit * nits * fix gather/scatter * fix rebase * reorg compile * add ternary to compile * jit copy * jit compile flag * fix build * use linked function for ternary * some nits * docs + circle min size build * docs + circle min size build * fix extension * fix no cpu build * improve includes	2024-05-22 12:57:13 -07:00
Awni Hannun	863039da4c	Allow scatter type exception to be caught by checking in op (#1077) ... * allow exception to be caught in main thread * only for gpu * more detailed scatter error	2024-05-13 17:43:53 -07:00
Nripesh Niketan	a30e7ed2da	feat: metal formatting and pre-commit bump (#1038) ... * feat: metal formatting and pre-commit bump * add guards * update * more guards * more guards * smakk fix * Refactor instantiation of ternary types in ternary.metal * fix scan.metal	2024-04-30 07:18:09 -07:00
Jagrit Digani	b219d12a6b	Check edge case handling in row reduce med kernel (#858)	2024-03-20 11:37:58 -07:00
Jagrit Digani	8dfc376c00	Strided reduce specialization for small reductions (#826) ... * Add small column / general reduction specialization	2024-03-14 09:16:53 -07:00
Jagrit Digani	6686e61ca4	Reduce update (#783) ... * Split reduction files to reduce compile times * Add small and medium axis size specializations for row reductions * Add non-row-reduction options for small and med kernels	2024-03-04 19:09:51 -08:00