Skip to content

[rocm]use aiter triton kernel as triton mha fallback path#809

Draft
zhuyuhua-v wants to merge 2 commits into
dev/perffrom
yuhua/triton-mla-fallback
Draft

[rocm]use aiter triton kernel as triton mha fallback path#809
zhuyuhua-v wants to merge 2 commits into
dev/perffrom
yuhua/triton-mla-fallback

update ROCM condition in flash attention

58a25ff
Select commit
Loading
Failed to load commit list.