Ivan Komarov f692b98d80 Fix spurious re-compilations of rotary_kernel (#911) ... All integer parameters are specialized by default, so the two parameters removed in this commit could lead to kernel re-compilation, even if they were completely unused.		2024-04-05 13:40:41 -07:00
..
triton	Fix spurious re-compilations of rotary_kernel (#911)	2024-04-05 13:40:41 -07:00
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
activations.py	[MLP] Implement SwiGLU with torch jiterator	2023-09-04 15:43:53 -07:00
fused_dense.py	[bugfix] handle_x not define when using checkpoint_lvl = 2 (#502)	2023-08-29 23:46:10 -07:00
layer_norm.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
rms_norm.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00