Skip to content

Atrex Kernel Agent v0.2.0 — Orchestrated optimization loop

Latest

Choose a tag to compare

@smallmou smallmou released this 18 Jul 12:03

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-optimizer Skill + 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.py owns 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.