Update gluon

[fsx950223]

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[github-actions]

🏷️ CI Guide

Runs automatically on every PR:

• ✅ Pre-checks (submodule verification, code formatting)
• ✅ Aiter op tests (gfx942 + gfx950)
• ✅ Triton tests on MI35X (only when aiter/ops/triton/** or related paths are changed)

Extended tests (opt-in via labels):

Label	Tests
ci:triton-300x	Run an additional Triton test job on MI300X in PRs; main branch always runs both MI35X and MI300X
ci:sglang	SGLang integration tests: DeepSeek-R1-MXFP4 accuracy, Qwen 3.5 accuracy
ci:atom	ATOM benchmark: DeepSeek-R1-0528, GPT-OSS-120B
ci:atom_full	ATOM accuracy suite for PR and main models from ATOM models_accuracy.json
ci:vllm	vLLM benchmark: GPT-OSS-120B, DeepSeek-R1-0528, Kimi-K2.5
ci:all	All standard extended tests (excludes ci:atom_full)

Only add ci:atom_full for FlyDSL or Triton upgrades.
Add labels via the sidebar or gh pr edit 3114 --add-label <label>

[Copilot]

Pull request overview

This PR updates the Gluon-based paged-attention decode implementation, including AOT compilation metadata and PS (partitioned softmax) reduction paths, to better support FP8 workflows and configurable output/temporary dtypes.

Changes:

• Add OUTPUT_DTYPE constexpr plumbed through Gluon decode kernels and the wrapper, and adjust temporary output dtype defaults for FP8 queries.
• Update PS reduce FlyDSL kernel codegen to newer FlyDSL APIs (fx.*, const_expr) and adjust reduction fallback ordering.
• Extend Gluon AOT compile signature/function-keying to include output and temporary output dtypes.

Reviewed changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.

File	Description
op_tests/triton_tests/test_pa_decode_gluon.py	Runs more local test entrypoints when invoked as a script (__main__).
csrc/cpp_itfs/pa_gluon_aot/pa_decode_gluon_aot.py	Adds output/temporary dtype specialization to AOT compile signature and naming.
aiter/ops/triton/gluon/pa_decode_gluon.py	Adds OUTPUT_DTYPE to kernels/wrappers, updates FP8 query handling, and refactors PS reduce fallback logic/FlyDSL kernel.

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.