KLineage: Apprendere i Tempi di Ottimizzazione da Kernel GPU Esperti

Un nuovo preprint arXiv (2605.28213) introduce KLineage, un sistema che insegna ad agenti basati su LLM quando applicare ottimizzazioni di kernel GPU apprendendo da implementazioni esperte. Invece di rollout forward, KLineage percorre i kernel esperti all'indietro attraverso semplificazioni convalidabili, invertendo ogni passo accettato in un'abilità riutilizzabile. Ogni abilità registra l'intento di ottimizzazione, la posizione nel codice, le condizioni di validità, gli effetti e le potenziali modalità di fallimento. Un LLM a valle applica queste abilità a nuovo codice sotto vincoli di compilazione/correttezza/profilo. Su cinque carichi di lavoro esperti su due architetture NVIDIA, le abilità derivate dalla lineage hanno superato le baseline LLM-kernel basate su memoria recente in termini di qualità finale del kernel.