arXiv 2606.02963v1Jun 1, 2026

KForge: LLM-Driven Cross-Platform Kernel Generation for AI Accelerators

Taras Sereda et al.

Brief context

Publication timing, weekly edition context, and source links for this brief.

Published

Jun 1, 2026, 11:48 PM

Current score

75

Original paper

The executive brief below is grounded in the source paper and linked back to the arXiv abstract.

Production inference increasingly targets a heterogeneous mix of accelerators. Agentic pipelines interleave reasoning, tool calls, and multi-agent coordination, each with distinct compute and memory profiles. For optimal efficiency, each stage should run on the accelerator best suited to it. This creates a systems challenge: each pipeline now requires high-performance kernels across a growing set of hardware backends and programming models. Writing these kernels by hand is time-consuming, demands deep low-level expertise, and does not scale as kernel complexity grows. Recently, Large Language Models (LLMs) have been leveraged for automatic kernel generation, but challenges in low-level code generation and cross-backend generalization persist. We present KForge, a cross-platform framework built around an iterative refinement loop driven by two collaborating LLM-based agents: a generation agent that produces and progressively refines kernels using compilation and correctness feedback, and a performance-analysis agent that interprets profiling data, from programmatic APIs to GUI-based tools, and emits recommendations that steer the next round of synthesis. The loop alternates between functional passes, which drive a candidate to correctness, and optimization passes, which close the performance gap to hand-tuned baselines. We evaluate KForge on two backends with very different baseline reference availability. On NVIDIA B200, KForge achieves a 2.12$\%$ improvement in end-to-end throughput compared to TensorRT-LLM on the gpt-oss-20b inference speed benchmark. On Intel Arc B580, KForge generates Triton kernels achieving a 5.13$\times$ geometric mean speedup over the faster of PyTorch eager and torch.compile on 37 GEMM + tail-ops workloads from KernelBench Level 2, primarily via operator fusion and mixed-precision execution.

Score 75Full-paper briefinferenceinfraagentsmodels

Executive brief

A short business-reader brief that explains why the paper matters now and what to watch or do next.

Why this is worth your attention

Kernel engineering is becoming a bottleneck in AI infrastructure strategy: every new accelerator choice creates a new pile of low-level code to write, tune, and maintain. This paper shows a credible path to making that work partially machine-generated, with small end-to-end gains over TensorRT-LLM on NVIDIA B200 and much larger benchmark gains on Intel Arc B580 where the software stack is less mature. If the pattern generalizes, infrastructure and procurement teams get more leverage in heterogeneous accelerator planning; what remains uncertain is whether these gains survive broader workloads, closed-source vendor kernels, and production tuning complexity.

  • If this approach holds up, teams running mixed accelerator fleets would not need to wait for every vendor library or hire rare low-level specialists for each backend. The business implication is faster hardware optionality: NVIDIA, AMD, Intel, and Apple silicon can be evaluated on workload fit rather than dismissed because the kernel work is too bespoke.
  • The NVIDIA result is not a headline-grabbing leap, but a 2.12% end-to-end throughput gain over TensorRT-LLM on B200 is meaningful in high-volume inference economics. It suggests LLM-generated kernels may find workload-specific savings even inside already-optimized stacks.
  • The Intel Arc B580 result is the more strategic signal: 5.13× geometric-mean speedup across 37 GEMM-plus-tail workloads came largely from fusion and mixed precision. If similar automation lands in production toolchains, non-dominant accelerators could compete better for specific inference stages where software support has been the blocker.
  • For procurement or platform evaluation, the hard question is whether generated kernels come with guardrails, reproducible artifacts, correctness checks, profiling traces, and end-to-end measurement hooks. A flashy auto-generated microbenchmark is much less valuable than a controlled loop that can be audited, repeated, and tied to production latency or throughput.
  • The evidence is encouraging but narrow: two accelerator case studies, PyTorch-oriented workflows, source-level kernels, and careful benchmarking controls. The paper itself notes that isolated kernel gains may not survive whole-system interactions, so the adoption test is sustained end-to-end improvement under normal production tuning conditions.

Evidence ledger

The strongest claims in the brief, along with the confidence and citation depth behind them.

capabilityhighp.1p.2

KForge uses two LLM-based agents to iteratively generate, correct, profile, and optimize accelerator kernels.

inferencehighp.5

On NVIDIA B200, KForge reports a 2.12% end-to-end throughput improvement over TensorRT-LLM on the gpt-oss-20b inference benchmark.

capabilityhighp.5

On Intel Arc B580, KForge-generated Triton kernels report a 5.13× geometric-mean speedup across 37 KernelBench Level 2 workloads.

stackhighp.2

The framework is designed for cross-platform kernel generation across four accelerator vendors and six programming models.

Related briefs

More plain-English summaries from the archive with nearby topics or operator relevance.

cs.CL

DeferMem: Query-Time Evidence Distillation via Reinforcement Learning for Long-Term Memory QA

Jianing Yin, Tan Tang

cs.DC

Compliance-Scored Best-of-N Guardrail Orchestration for Multimodal Document Generation in Payments Dispute Defense

Nataraj Agaram Sundar, Tejas Morabia

cs.CL

SynAE: A Framework for Measuring the Quality of Synthetic Data for Tool-Calling Agent Evaluations

Shuaiqi Wang et al.

cs.LG

FlowBank: Query-Adaptive Agentic Workflows Optimization through Precompute-and-Reuse

Lingzhi Yuan et al.

Thank you to arXiv for use of its open access interoperability. This product was not reviewed or approved by, nor does it necessarily express or reflect the policies or opinions of, arXiv.
LightDark