4 Commits

Author	SHA1	Message	Date
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