Atrex Kernel Agent(AKA) v0.2.0
This release grows AKA from a single interactive skill into a dual-route, profile-driven kernel optimization system, adds a full NVIDIA profiling toolchain, and significantly expands the GPU knowledge base.
Highlights
- Two optimization routes. The same profile-driven workflow can now be driven either interactively (Route 1:
gpu-kernel-optimizerSkill + hooks) or unattended (Route 2:orchestrator/optimize.py). Both share the same knowledge base, tools, and structured memory format. - Orchestrated loop (Route 2).
orchestrator/optimize.pyowns the outer loop, spawning a fresh clean session per iteration over a single git workspace. State crosses sessions only through disk, HEAD is always the best kernel, and termination is mechanical (max iterations / token budget / target-utilization short-circuit). - SOL-ExecBench native. Drive a single-operator campaign directly from a SOL-ExecBench op directory (
definition.json,reference.py,workload.jsonl) via--op-dir; ground-truth files are used verbatim and never edited.
Features
- Clean-session-per-iteration optimization loop with git-native, restart-safe stall tracking.
- Automated Triton→Gluon convert session (runs inline, gated by a stall counter via
--convert-after). - Subagent architecture under
agents/(baseline, profiler, research decompose, convert, partial-restart) shared by both routes. - Decompose step for fusion ops.
- Stall-count gated search strategy in the Stage-2 research phase.
- NVIDIA profiling toolchain:
tools/profile_nvidia.sh,tools/ncu_helpers/,tools/classify_ncu.py,tools/extract_nvidia_asm.py,tools/extract_ttgir.py. - Submodule-based reference management for
reference-projects/and 3rdparty knowledge/tooling.