automatic/modules/flash_attn_triton_amd
Seunghoon Lee 18f58e13f8
Update fwd_prefill.py
2025-05-09 13:36:01 +09:00
..
__init__.py zluda flash attention 2 via triton 2025-03-22 23:10:59 +09:00
fwd_prefill.py Update fwd_prefill.py 2025-05-09 13:36:01 +09:00
interface_fa.py zluda 3.9.4 & update flash attention 2 2025-05-07 00:39:40 +09:00
utils.py zluda 3.9.4 & update flash attention 2 2025-05-07 00:39:40 +09:00