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
..
blas	Update GEMM (#424)	2024-01-17 12:42:39 -08:00
comparative	feat: Update pre-commit-config.yaml (#667)	2024-02-11 06:08:20 -08:00
batch_matmul_bench.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
compile_bench.py	Shapeless compilation for some graphs (#687)	2024-02-19 21:43:54 -08:00
gather_bench.py	Scatter optimization : Eliminate 64b integer divide. (#662)	2024-02-10 08:49:51 -08:00
rope_bench.py	Custom primitive + RoPE fat op (#676)	2024-02-14 14:04:25 -08:00
scatter_bench.py	Scatter optimization : Eliminate 64b integer divide. (#662)	2024-02-10 08:49:51 -08:00
single_ops.py	Propagate nans in binary ops (#579)	2024-01-29 11:19:38 -08:00
time_utils.py	Shapeless compilation for some graphs (#687)	2024-02-19 21:43:54 -08:00