SueJane 3f1b4d38e7 Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127)		2024-08-05 08:59:23 -07:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
block.py	[LayerNorm] Switch from CUDA to Triton implementation	2024-01-05 00:31:17 -08:00
embedding.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
mha.py	Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127)	2024-08-05 08:59:23 -07:00
mlp.py	Add Alibi to MHA, test with Baichuan-13B	2023-12-21 22:49:55 -08:00