Skip to content

[FA2] Use kernels-community/aiter-flash-attn-ck Hub kernel for ROCm FlashAttention (drop aiter dependency) - #14436

Open
Abdennacer-Badaoui wants to merge 1 commit into
huggingface:mainfrom
Abdennacer-Badaoui:use-aiter-fa-from-hub
Open

[FA2] Use kernels-community/aiter-flash-attn-ck Hub kernel for ROCm FlashAttention (drop aiter dependency)#14436
Abdennacer-Badaoui wants to merge 1 commit into
huggingface:mainfrom
Abdennacer-Badaoui:use-aiter-fa-from-hub