Awni Hannun 5798256fcf Shapeless compilation for some graphs (#687) ... * shapeless compilation for some graphs * update compile benchmark * default compile a few activations * buffer donation * bugfix * shapeless fix * update tests to work for cpu and gpu fusion * test kwargs * add kwargs to compile * Recompile when python arguments change * no compile for tanh * some constant tests --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>		2024-02-19 21:43:54 -08:00
..
cpp	Multi output primitives (#330)	2024-01-08 16:39:08 -08:00
numpy	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
python	Shapeless compilation for some graphs (#687)	2024-02-19 21:43:54 -08:00