10 Commits

Author	SHA1	Message	Date
Awni Hannun	3f86399922	Real and Imag (#1490) ... * real and imag * fix * fix	2024-10-15 16:23:15 -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
Nripesh Niketan	669c27140d	Chore: add pre-commit hook for cmake (#1362) ... * reset and lint * format --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-09-16 12:53:01 -07:00
Angelos Katharopoulos	5c1fa64fb0	Custom transforms (#1246)	2024-07-10 18:00:01 -07:00
Alex Barron	a3c287354f	Fast Hadamard Transform (#1249) ... * Working hadamard for powers of 2 * working for m*2^k * add scale and check contiguity * add size check * clean up * fix test * add grads + vmap * gpu only * skip on linux * test typo * add cpu impl * remove gpu only tests * fix linux build + add is_equivalent	2024-07-09 20:39:01 -07:00
Awni Hannun	fe3167d7ea	smaller CPU binary (#1203) ... * smaller CPU binary * fix no cpu build	2024-06-14 09:46:55 -07:00
Alex Barron	375a8bbdcc	Add some internal GPU apis (#1177) ... * Add unary/binary/ternay/slice/concat internal GPU ops * add pad internal op * formatting + no_cpu fix	2024-06-04 09:24:26 -07:00
Awni Hannun	ea9090bbc4	Add view op (#1179) ... * add view primitive * nit * fix view	2024-06-04 08:05:27 -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	7178ac0111	No CPU option for binary minimization (#1105) ... * no cpu build option * docs * fix	2024-05-13 16:08:11 -07:00