6 Commits

Author	SHA1	Message	Date
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	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	4f72c66911	improvements to scatter / gather (#1541)	2024-10-30 19:30:54 -07:00
Awni Hannun	df3233454d	2d gather specialization (#1339)	2024-08-22 10:48:24 -07:00
Awni Hannun	df964132fb	fix scatter + test (#1202) ... * fix scatter + test * fix test warnings * fix metal validation	2024-06-11 14:35:12 -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