5 Commits

Author	SHA1	Message	Date
Awni Hannun	dc937b8ed3	CPU compile (#691) ... * build and load shared object for cpu compile * nits * cpu compile tests pass * cpu compile tests pass * fix preamble for g++ * donation * fix gpu buffer donation * reuse prebuilt libraries * faster contiguity conditoins * fix test * rid compiler warning * fast erf * Fix float16 for compile and add more types to cpu compile * Remove a forgotten comment * use cached libs * nits --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-02-17 06:54:32 -08:00
Angelos Katharopoulos	28eac18571	Kernel generation (#614) ... Generate reusable element-wise kernels given a computation graph.	2024-02-07 13:15:59 -08:00
Awni Hannun	146bd69470	Skip compile when transforming (#635) ... * skip compile when transforming * simplify message	2024-02-05 21:28:37 -08:00
Awni Hannun	d75ae52ecd	Compile primitive (#571) ... * Compiled primitive with basic binary, unary graph-level fusion	2024-02-05 06:51:22 -08:00
Awni Hannun	8fa6b322b9	Compile front-end (#476) ... * fix tests for linux * make a move on compile * basic compile scaffold works * compile binding * clean * fix * fix grad, more tests * basic python tests * fix segfault on python exit * compile works with python closures * fix test * fix python globals bug, and erase * simplify * more cpp tests * bug fix with move function and compile at exit * simplify inputs also * enable and disable compiler * remove simplify * simplify tests use compile now * fix multi-output with compile * clear output tree from cache when function goes out of scope * ../python/src/transforms.cpp * remove closure capture * comments	2024-01-26 13:45:30 -08:00