1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
automatic/modules/flash_attn_triton_amd
History
Seunghoon Lee ab9e87d848
zluda flash attention 2 via triton 		2025-03-22 23:10:59 +09:00
..
__init__.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
bwd_prefill.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
bwd_ref.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
fwd_decode.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
fwd_prefill.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
fwd_ref.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
interface_fa.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
utils.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00