Snilld

CUDA Agent gør LLM‑kernels konkurrencedygtige på fart

ByteDance Seed og Tsinghua AIR præsenterer CUDA Agent, et agentisk RL‑system der træner en stor sprogmodel til at skrive CUDA‑kernels, som på en 250‑opgavers benchmark oftere slår torch.compile end den taber – med 2,11× geomean‑speedup ifølge forskerholdets egen rapportering. Det lukker et konkret performance‑gab og har praktiske konsekvenser for teams med GPU‑tunge arbejdslaster, men fuld replikation kræver massiv infrastruktur, og vægtene er ikke frigivet.

18. august 2026 Peter Munkholm

CUDA Agent lander som et reelt fartløft på benchmark

ByteDance Seed og Tsinghua AIR har offentliggjort CUDA Agent, et agentisk reinforcement learning‑system der lærer en LLM at generere CUDA‑kernels, som ikke bare er korrekte – men hurtige. Ifølge gennemgangen på MarkTechPost løfter systemet ydeevnen markant på en 250‑opgavers benchmark: 98,8 procent korrekthed, hurtigere end eager i 98,4 procent af tilfældene og hurtigere end torch.compile i 96,8 procent, med en geometrisk middelværdi på 2,11× over compile.

Det kan lyde smalt, men er vigtigt. Basismodeller kan allerede skrive korrekt CUDA, blot ofte langsom CUDA. På KernelBench klarer Seed1.6 74,0 procent af opgaverne korrekt, men slår kun torch.compile i 27,2 procent – og ender i gennemsnit på 0,69× fart, altså langsommere end compileren. CUDA Agent retter sig mod præcis det hul.

Makrofoto af et måleprobe‑kontaktpunkt på en testplade med cyan‑grøn refleks; sløret GPU‑blade i baggrunden.

Hvad CUDA Agent egentlig er

Kort sagt sætter forskerholdet en stor sprogmodel ind i et rigtigt CUDA‑devmiljø med profilering, korrekthedstests og en tilladelseslåst sandbox. Træningen kører med Proximal Policy Optimization (PPO) i 150 skridt og en kontekstlængde på 131.072 tokens, ifølge MarkTechPost. Det hele bygger på Seed1.6 – en proprietær mixture‑of‑experts model, rapporteret til 23 mia. aktive parametre og 230 mia. i alt.

Vægtene til den trænede agent er ikke frigivet. Det udelukker et drop‑in. Men artefakterne omkring er frigivet: datasættet CUDA‑Agent‑Ops‑6K, SKILL.md‑specifikationen samt opskrifter til reward og warm‑up. Det er materiale, mellemstore teams faktisk kan bruge.

Agent‑loopen i praksis

Miljøet ligner kendte dev‑agenter: Bash, Read/Write, Edit/MultiEdit, Glob, Grep, NotebookEdit, BashOutput, KillBash – styret med et ReAct‑mønster. CUDA‑instruktioner pakkes i et Agent Skills‑format, og SKILL.md instruerer modellen i at profilere PyTorch‑grafen, skrive brugerdefinerede kernels i model_new.py, kompilere i en GPU‑sandbox og iterere, indtil koden er mindst 5 procent hurtigere end torch.compile (med atol=1e‑2, rtol=1e‑2).

Belønningen er diskret, ikke en rå speedup‑ratio. r ∈ {−1, 1, 2, 3}: −1 ved korrekthedsfejl, 1 for baseline, 2 hvis eager slås, 3 hvis både eager og torch.compile slås med >5 procent. Fem modtræk skal dæmme op for reward‑hacking: låste verificerings‑ og profileringsscripts, context managers der forhindrer fald tilbage til torch.nn.functional, test mod fem tilfældige input, profilering med device‑sync og warm‑up, samt ingen web‑værktøj. Tørt? Ja. Nødvendigt? Også ja.

Data og syntese

Datasættet kommer ikke ud af det blå. Holdet crawler reference‑operatorer fra torch og transformers. En LLM sampler op til fem operator‑klasser og stabler dem til én fused lag. Filtrene er stramme: kun operatorer der kører i både eager og compile, er deterministiske, giver ikke konstante outputs og ligger mellem 1 og 100 ms i eager‑tid. Prøver med AST‑lighed over 0,9 i forhold til KernelBench bliver fjernet.

Banner

Resultatet kaldes CUDA‑Agent‑Ops‑6K: 6.000 prøver, hvor 83,77 procent er to‑operatorsammensætninger. Snævert, ja, men fokuseret mod de sekvenser, hvor torch.compile ikke altid fuser optimalt.

Et lille PoC‑testsetup i et laboratorium: en operationsleder (uden ansigt i frame) peger på et rødt/indigo markeret pipelinekort på væggen mens en tekniker justerer en testrig, cyan highlights i rummet.

Resultaterne og en lille parentes

Samlet set rapporterer holdet 98,8 procent pass rate, 98,4 procent hurtigere end eager og 96,8 procent hurtigere end torch.compile – med 2,60× og 2,11× geomean‑speedups. Niveau 2 (operatorsekvenser) er stærkest: 100 procent pass, 100 procent hurtigere rate, 2,80× over compile. Niveau 3 lander på 94,0 procent pass, 90,0 procent hurtigere rate og 1,52×. Ifølge artiklen ligger det omtrent 40 point over Claude Opus 4.5 og Gemini 3 Pro på hurtigere‑end‑compile‑raten på den sværeste split.

En inkonsistens nævnes direkte: Abstract og indledning taler om 100/100/92 procent for Levels 1–3, mens tabel 1 siger 97,0/100,0/90,0. Artiklen peger selv på tabel 1 som den primære reference. Det er værd at få frem, før nogen oversælger.

Hvad der faktisk lukker performance‑gabet

Tre ting skiller sig ud. 1) Den tætte kobling til en profilerings‑realitet frem for syntetisk belønning. 2) Et diskret belønningsdesign med klare tærskler og anti‑snyd‑værn, så modellen ikke optimerer sig uden om målet. 3) Et datasæt og opgaveformuleringer, der fremtvinger kernel‑fusion og memory‑bevidste mønstre. Disciplineret ingeniørarbejde – og tallene afspejler det.

Ressourcer og reproduktion

Ifølge MarkTechPost kørte profilerings‑sandboxen på 128 NVIDIA H20‑GPU’er. Det siger to ting: evalueringen er tung, og fuld replikation er dyr. For laboratorier, hyperscalere og GPU‑skyer er det hverdag; for alle andre et benspænd. Samtidig er vægtene ikke frigivet, så man kan ikke uafhængigt validere de præcise geomean‑speedups.

Det gør artefakterne desto vigtigere. Datasættet, SKILL.md og reward/warm‑up‑opskrifter kan flyttes over på en åben basismodel og give dele af gevinsten. Ikke hele pakken – men nok til at rykke latenstid, hvis man vælger sine workloads klogt.

Makrofoto af et måleprobe‑kontaktpunkt på en testplade med cyan‑grøn refleks; sløret GPU‑blade i baggrunden.

Hvem der får mest ud af det

Brugsscenarierne er velkendte: AI‑infrastruktur og inference‑serving, GPU‑clouds, autonome systemer, kvantitativ handel, medicinsk billeddiagnostik og anbefalingssystemer. Steder hvor fused kernels afgør latenstiden, og hvor torch.compile ikke altid rammer plet. Tænk token‑trin i store modeller, batchede attention‑varianter eller små, varme opgaver i recommender‑pipelinen.

Hvis man måler hårdt på cost‑per‑token og p99‑latenstid, er muligheden reel. Ikke romantisk. Reel. En håndfuld millisekunder pr. kald bliver hurtigt til femcifrede beløb pr. måned i drift.

Hvad mellemstore teams konkret kan genbruge

Tre oplagte dele: 1) CUDA‑Agent‑Ops‑6K som trænings-/eval‑sæt, 2) SKILL.md som metode til at instruere en model i performance‑arbejdsgange, 3) belønnings‑ og warm‑up‑opskrifter inklusive anti‑reward‑hacking‑tjek. Oven på en åben basismodel kan det give målbare forbedringer uden adgang til 128 H20‑kort.

Virkeligheden er dog, at man behøver profileringsinfrastruktur, deterministiske testinputs, device‑sync i målinger, CI‑jobs der kører korrekthed og performance side om side, samt sandboxes med stramme tilladelser. Det er ikke “kør model og deploy”. Ofte er GPU‑cloud til PoC eller en partneraftale den hurtigste vej frem.

Banner

Drift og governance bør ikke vente

Enterprise‑organisationer får samtidig et styringsproblem. Agent‑sprawl er reel, og styringslaget halter ofte bagefter. VentureBeat fremhæver, at store virksomheder risikerer hundredtusinder af agenter uden ordentlig kontrol med tilladelser, observabilitet og livscyklus. Det gælder også for agenter, der må pille ved kernels.

Det praktiske svar: rollebaseret adgang til build‑ og deploy‑trin, audit‑logs på genererede artefakter, faste performance‑guardrails og en rullende releaseproces med canary og automatisk rollback, når en ny kernel ikke holder i produktion. Og ja – behold fallback‑stier til torch.compile i alle paths.

Risici og begrænsninger

Det største spørgsmål er generalisering. Benchmarket er stærkt, men resultaterne kommer fra forskerholdets egen sandbox, uden peer review eller uafhængig tredjepartsverifikation. Der er også hardware‑bias: profileringsvalg og cache‑karakteristik kan ændre rangordenen mellem strategier fra én GPU‑generation til den næste.

Datasætteffekter er værd at nævne. Filtrene (1–100 ms, determinisme, AST‑fjernelse ved 0,9) kan uundgåeligt vinkle opgaverne mod steder, hvor agenten har lettest ved at vinde. Det er ikke en kritik, bare en grund til at køre egne workloads igennem, før man lover besparelser. Og igen: uden udgivne vægte kan man ikke genskabe kurverne én til én.

Konkurrencesituationen

Det interessante er, at agentisk RL med tæt profilering ser ud til at flytte mere end promptteknikker eller ren supervised finetuning. Det er en klar faglig pointe, uanset brand.

Næste skridt for tekniske beslutningstagere

Et nøgternt PoC kan se sådan ud: vælg 10–20 latency‑kritiske operatorsekvenser fra jeres graf, mål baseline i eager og torch.compile med device‑sync, brug SKILL.md‑mønstret og et simpelt diskret reward‑skema, finetun en åben model på et subset af CUDA‑Agent‑Ops‑6K og jeres egne samples, og mål så geomean‑speedup mod compile. Pas især på forvarmning, batchstørrelse og input‑distribution – tre klassiske faldgruber.

Succesmål bør være jordnære: 15–30 procent hurtigere end torch.compile på mindst halvdelen af PoC‑opgaverne, stabil korrekthed på ≥99 procent over fem tilfældige input og ingen regressioner på produktionslignende data. Ser I det, er værdien reel. Hvis ikke, så parker – eller stram datasæt og belønning.

Hvem man kan ringe til for kommentar

Fagligt blik udefra er nyttigt. Relevante kilder inkluderer forskere fra ByteDance Seed og Tsinghua AIR, GPU‑cloud‑udbydere med profileringsløsninger, MLOps‑værktøjsleverandører der arbejder med agent‑governance og enterprise‑IT‑arkitekter, der driver performancekritiske inferensmiljøer. Korte interviews kan afklare, hvor meget af gevinsten der skyldes datasættet, belønningsdesignet – eller bare færre fejl i måleopsætningen end konkurrenterne.

Og husk – der er ingen skam i at lade andre skyde hullerne først. Det sparer tid.

Vurdering

CUDA Agent er et teknisk gennembrud på det benchmarket, som holdet selv beskriver. Der er et klart fartløft over torch.compile, og metoden – agentisk RL inde i et rigtigt dev‑miljø – er solidt tænkt. Men det er ikke et drop‑in til alle på mandag. Manglende vægte, tung replikations‑infrastruktur og uklare generaliseringsgrænser taler for målrettede PoC’er frem for storladne roadmap‑løfter.

Hvad er frigivet, og hvad mangler

Offentligt tilgængeligt ifølge MarkTechPost: datasættet CUDA‑Agent‑Ops‑6K, SKILL.md‑specifikationen og opskrifter til reward og warm‑up. Ikke frigivet: den trænede agents vægte. Det gør fuld reproducerbarhed svær og holder den præcise hastighedsfordel i laboratoriet lidt endnu.

Praktisk konsekvens: vælg jeres kampe. Start småt, mål hårdt, og bevar compile‑fallbacks. Forskellen opdages først, når man har det i hænderne.

Kilder

    Gør brugeroplevelsen bedre.
    Hvilket firma arbejder du for?