feat(gdn): add BF16 state kernel with MTP support beyond T>4 with intermediate caching.#2679

Merged

kahyunnam merged 13 commits intoflashinfer-ai:mainfrom

ameynaik-hub:ameyn/gdn_bf16_improvements

Apr 2, 2026

Commits on Mar 3, 2026

feat(gdn): add BF16 state kernel with MTP support

ameynaik-hub
and
claude
committed

Commits on Mar 18, 2026

merge main into ameyn/gdn_bf16_improvements, resolve conflicts

ameynaik-hub
and
claude
committed
Merge branch 'main' into ameyn/gdn_bf16_improvements
ameynaik-hub
committed

Commits on Mar 19, 2026

fix(bench): accurate CUPTI timing for BF16 state MTP benchmark

ameynaik-hub
and
claude
committed

Commits on Mar 21, 2026

Merge branch 'main' into ameyn/gdn_bf16_improvements

ameynaik-hub
and
claude
committed

Commits on Mar 23, 2026

Merge branch 'main' into ameyn/gdn_bf16_improvements
ameynaik-hub
committed

Commits on Mar 24, 2026

Commits on Mar 27, 2026

fix: add scalar params to cute.compile() cache keys, unskip tests, add intermediate state checks

ameynaik-hub
and
claude
committed

Commits on Mar 31, 2026

Commits on Apr 1, 2026

fix: replace assert_close with manual comparison for large intermediate state tensors

ameynaik-hub
and
claude
committed

Commits on Apr 2, 2026

fix type annotation
ameynaik-hub
committed