Integrate DS R1 GroupedTopk + Sigmoid Routing Into DS Routing

[amirumoAMD]

Motivation

Integrate new GroupedTopk kernel with support for score_mode sigmoid into routing + unified routing function, renamed function to reflect change. For changes done in ROCm/ATOM#1044.

Technical Details

grouped_topk kernel added to topk files, integrated into grouped topk.

Test Plan

Pytest for kernel included in routing pytest file, lm_eval on atom deepseek r1 mxfp4

Test Result

Kernel pytests passing, lm_eval 98% for atom integration for deepseek r1 mxfp4.

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 3522 --add-label <label>

[k50112113]

Can you also help check if the _grouped_topk kernel you added here is functionally the same as the one on the HIP side? no need to add unit test

fixed recommended issues, dismissing stale review