8 Commits

Author	SHA1	Message	Date
Awni Hannun	516ded618b	Dynamic slicing (#1741) ... * dynamic slice and slice update * python bindings + tests + fix set item * fix compile issue * comment * fix jit	2025-01-07 14:02:16 -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	211411faf2	fix large ops (#1620)	2024-11-24 09:17:10 -08:00
Awni Hannun	2419edd5b2	Faster indexing math in a few kernels (#1589) ... * wip: faster compiled kernels * faster general unary with uint specialization * index type in compiled, unary, binary, ternary, copy * fix jit * jit fix * specialize gather + scatter * nit in docs	2024-11-18 19:52:00 -08:00
Awni Hannun	67b6bf530d	Optimization for general ND copies (#1421)	2024-09-17 17:59:51 -07:00
Awni Hannun	4f46e9c997	More fixes for arrays with large sizes (#1405) ... * compile works for big arrays when contiguous * style * nits in docs * a bunch more stuff * update jit * update jit * use constant for shapes and strides and remove elem_to_loc overload * use kernel instantiation * docs nits * update binary and ternary * comments	2024-09-17 12:46:31 -07:00
Awni Hannun	40b6d67333	Fixes for large arrays with a few ops (#1299) ... * fixes for large arrays with a few ops * fix bug * fix all of copy	2024-07-30 17:18:39 -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