fix(cake_kda): support non-aligned recurrent prefill head counts - #4351

fix(cake_kda): support non-aligned recurrent prefill head counts#4351

yyihuang merged 1 commit into

flashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

yyihuang:codex/kda-prefill-h12-exportyyihuang/flashinfer:codex/kda-prefill-h12-exportCopy head branch name to clipboard