Felix Marty 38447262c0 fix fused attn		2023-07-31 13:46:32 +00:00
..
qlinear	fix fused attn	2023-07-31 13:46:32 +00:00
triton_utils	support 32dim triton kernel	2023-06-02 19:04:12 +09: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	fix fused attn	2023-07-31 13:46:32 +00:00
fused_llama_attn.py	fix fused attn	2023-07-31 13:46:32 +00:00
fused_llama_mlp.py	update FusedLlamaMLPForQuantizedModel for general usage purpose	2023-05-27 07:47:20 +08:00