Make FlyDSL LDS checks architecture-aware and reduce tuner failure noise

[yzhou103]

Motivation

Technical Details

• Use device shared-memory/LDS limits for FlyDSL GEMM filtering and split-k kernel validation instead of relying on a fixed LDS cap.
• Surface FlyDSL candidate compile/LDS failures as concise runtime warnings so tuning can continue without noisy tracebacks.
• Keep tuner topk selection local to each shape to avoid leaking a reduced candidate limit into later shape groups.

Test Plan

• Verified failing candidates now log a single concise warning with kernel/context information and no extra traceback noise.
• Verified tuning still completes and produces a valid best kernel result.

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 (only when aiter/ops/triton/** or related paths are changed)

Extended tests (opt-in via labels):

Label	Tests
ci:triton-355	Run Triton tests on MI355 in addition to MI325
ci:sglang	SGLang integration tests
ci:atom	ATOM benchmark (DeepSeek-R1 + GPT-OSS)
ci:vllm	vLLM benchmark
ci:all	All of the above

Add labels via the sidebar or gh pr edit 2732 --add-label <label>

[Copilot]

Pull request overview

This PR makes FlyDSL tuning and kernel validation architecture-aware by using device-specific LDS/shared-memory limits, improves error handling to reduce noisy failures during tuning, and ensures topk selection doesn’t leak across shape groups.

Changes:

• Add a utility to query per-device shared memory/LDS limits (with arch-based fallback) and use it in FlyDSL LDS validation.
• Improve FlyDSL HGEMM kernel compilation error messages and make tuner workers bail out early on invalid results.
• Keep topk selection local per shape group during post-processing.

Reviewed changes

Copilot reviewed 6 out of 6 changed files in this pull request and generated 2 comments.

Show a summary per file

File	Description
aiter/utility/mp_tuner.py	Adds helper error classifiers; broadens handled exceptions and returns early for failed runs to keep tuning moving.
aiter/utility/base_tuner.py	Uses an effective_topk per shape group to prevent cross-group mutation of the global topk.
aiter/ops/flydsl/utils.py	Introduces shared-memory/LDS limit helpers using device properties with an arch fallback.
aiter/ops/flydsl/kernels/splitk_hgemm.py	Replaces fixed LDS assert with device-aware limit checks and wraps compile failures with concise contextual errors.
aiter/ops/flydsl/gemm_tune/flydsl_gemm_a8w8_bpreshuffle_common.py	Switches tuning LDS cap to device-aware helper and de-duplicates arch mapping logic.
aiter/ops/flydsl/gemm_kernels.py	Replaces the fixed MAX_LDS_BYTES cap with device-aware LDS limit validation.

---

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