PanQiWei 2b532f9453 add trainable mode		2023-05-26 13:11:30 +08:00
..
qlinear	add trainable mode	2023-05-26 13:11:30 +08:00
triton_utils	add trainable mode	2023-05-26 13:11:30 +08:00
__init__.py	refactor file structure	2023-04-25 18:58:20 +08:00
_fused_base.py	add trainable mode	2023-05-26 13:11:30 +08:00
fused_gptj_attn.py	add trainable mode	2023-05-26 13:11:30 +08:00
fused_llama_attn.py	add trainable mode	2023-05-26 13:11:30 +08:00
fused_llama_mlp.py	fix not return directly when triton is not installed	2023-05-20 16:21:52 +08:00