[Example] Add GQA decoding kernel with varlen page table (#1265)

* [Example] Add page table for gqa decode * [Example] Page table for varlen decoding * [Lint] * [Refactor] Remove redundant code * [Lint] * [Lint] * [Lint]

[Example] Add GQA decoding kernel with varlen page table (#1265)
* [Example] Add page table for gqa decode * [Example] Page table for varlen decoding * [Lint] * [Refactor] Remove redundant code * [Lint] * [Lint] * [Lint]
716dbef5 · Zhengju Tang · GitHub · 2de566e7 · 716dbef5
Unverified Commit 716dbef5 authored Nov 17, 2025 by Zhengju Tang Committed by GitHub Nov 17, 2025
Expand all Show whitespace changes
Inline Side-by-side

Showing with 711 additions and 0 deletions

examples/flash_decoding/example_gqa_decode_varlen_logits_paged.py .../flash_decoding/example_gqa_decode_varlen_logits_paged.py +711 -0

No files found.
--- a/examples/flash_decoding/example_gqa_decode_varlen_logits_paged.py
+++ b/examples/flash_decoding/example_gqa_decode_varlen_logits_paged.py