 mukesh
|
ddee28a499
|
gqa(decode-4cases): rename bench/kernels/panels with long_ctx token
The 4-cases comparative study is specifically about long-context
decode (LLaMA-3.1-70B, S_kv=128K, T_q=1); the long_ctx token in the
names makes the scope explicit and aligns with the existing
_gqa_attention_decode_long convention.
Renames (mechanical; no semantic change):
- bench file: milestone_gqa_decode_4cases.py
→ milestone_gqa_decode_long_ctx_4cases.py
- bench name: milestone-gqa-decode-4cases
→ milestone-gqa-decode-long-ctx-4cases
- output dir: 1H_milestone_output/gqa_decode_4cases/
→ 1H_milestone_output/gqa_decode_long_ctx_4cases/
- env vars: GQA_DECODE_4CASES_RUN / _TOPOLOGY
→ GQA_DECODE_LONG_CTX_4CASES_RUN / _TOPOLOGY
- 4 kernel files _gqa_attention_decode_<case>.py
→ _gqa_attention_decode_long_ctx_<case>.py
- 4 kernel functions gqa_attention_decode_<case>_kernel
→ gqa_attention_decode_long_ctx_<case>_kernel
- 4 dispatch kinds decode_<case> → decode_long_ctx_<case>
- 4 panel names single_kv_group_decode_gqa_<case>
→ single_kv_group_decode_long_ctx_gqa_<case>
- 4 helper functions _run_decode_panel_<case>
→ _run_decode_panel_long_ctx_<case>
- test file renamed in lockstep
Also: Case 4 smoke test now uses its case-specific helper
_run_decode_panel_long_ctx_cube_sp_pe_sp (consistent with Cases 1-3)
instead of the legacy _run_decode_panel from milestone_gqa_headline.
16 tests pass.
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
|
2026-06-15 16:09:21 -07:00 |
|