#34461 [ROCm] Extend the gfx950 extend-attention tile to head_dim <= 128: -43% kernel, -14% TTFT, bit-identical

原始 PR 作者 alexnails 合并时间 2026-08-24 15:29 文件变更 2 提交数 1 评论 1 代码增减 +17 / -16

执行摘要

测试与验证
- test/registered/attention/test_triton_attention_kernels.py（test-coverage）：测试配套；包含测试覆盖调整、控制流调整、配置键调整；+12/-10
配置与运行环境
- python/sglang/kernels/ops/attention/extend_attention.py（infrastructure）：部署/基础设施；包含控制流调整；+5/-6

文件	模块	状态	重要度
`test/registered/attention/test_triton_attention_kernels.py`	triton/attention/kernels	modified	5.16
`python/sglang/kernels/ops/attention/extend_attention.py`	sglang/kernels	modified	3.72

分析完成后，这里会展示 LLM 生成的相对完整源码片段和详细注释。

当前评论区没有形成足够清晰的争议点或结论，后续有更多讨论时会体现在这里。

当前没有检测到明确关联的 Issue 链接，后续同步到相关引用后会出现在这里。