Chiosa
IT
AI & Agents

Un supervisore aiuta i team di agenti a ripensare i kernel GPU

4 min
Aggiornato

In parole semplici

Il paper presenta Evolving Agent Teams, un modo per organizzare agenti di programmazione che scrivono programmi per GPU. Un supervisore ne segue i progressi e può cambiare i ruoli e il modo in cui gli agenti condividono le informazioni.

Un file system condiviso trasferisce le conoscenze e il lavoro da una configurazione del team alla successiva. Gli autori hanno testato il sistema su quattro attività di ottimizzazione di GPU kernel.

A parità di budget di token, i team adattivi hanno superato un singolo agente di programmazione in ciascuna attività. Questo non significa però che siano stati più veloci su ogni singolo carico di lavoro.

La ricerca

Quando gli agenti di programmazione lavorano su un GPU kernel difficile, possono fermarsi a una soluzione più semplice e smettere di esplorare alternative più impegnative. Cambiare il team man mano che il lavoro procede può aiutarlo a superare questo stallo? Il paper esamina la domanda su quattro attività di ottimizzazione di kernel.

Evolving Agent Teams usa un supervisore per riorganizzare ruoli e coordinamento degli agenti mentre ottimizzano GPU kernel, basandosi su registrazioni persistenti per guidare ogni nuova configurazione del team.

Gli autori hanno confrontato Evolving Agent Teams con Claude Code, un singolo agente di programmazione, su quattro attività di ottimizzazione di GPU kernel. Entrambi hanno ricevuto la stessa descrizione dell’attività, lo stesso kernel di riferimento, lo stesso sistema di test e lo stesso budget di token; per ciascuna configurazione è stato riportato il migliore di due tentativi.

Il team adattivo ha superato Claude Code in tutte e quattro le attività. Su MLA paged decode, un’esecuzione con budget aggiuntivo ha raggiunto uno speedup medio geometrico di 1.21x rispetto a FlashInfer in produzione, su 47 carichi di lavoro di produzione.

La misura combina i rapporti tra la latenza della baseline e quella del kernel candidato, quindi un valore superiore a 1 indica prestazioni aggregate più rapide. La conclusione supportata è circoscritta: secondo questa misura, il team ha prodotto un kernel più veloce, ma non su ogni singolo carico di lavoro.

Le configurazioni principali sono state eseguite due volte e il paper riporta il risultato migliore di ciascuna coppia. Alcuni carichi di lavoro possono comunque essere più lenti rispetto alla baseline di produzione, e il numero di esecuzioni limita le conclusioni statistiche.

Cosa mostrano le prove

  • Il rischio di fermarsi a una soluzione semplice

    Evolving Agent Teams è un framework per ottimizzare GPU kernel con agenti di programmazione. Il paper indaga se un team in grado di cambiare organizzazione durante un’attività possa affrontare una difficoltà ricorrente: gli agenti possono fermarsi a una soluzione più semplice e smettere di esplorare alternative più impegnative.

    Gli autori illustrano questo schema nell’ottimizzazione dei kernel, quindi i risultati riguardano queste attività e non gli agenti di programmazione in generale.

  • Un team che si adatta

    Il framework affida a un supervisore il compito di monitorare i progressi del team e individuare possibili stalli. Il supervisore può quindi modificare i ruoli, il coordinamento tra gli agenti o il modo in cui registrano e condividono le informazioni.

    Un file system persistente trasferisce osservazioni e materiali precedenti alle configurazioni successive del team, collegando i cambiamenti al lavoro già svolto.

  • Il confronto principale

    La valutazione principale ha riguardato quattro attività di ottimizzazione di GPU kernel su GPU B200 e H100. A parità di budget di token, Evolving Agent Teams ha superato Claude Code, un singolo agente di programmazione, in ciascuna attività.

    Per ogni configurazione gli autori riportano il migliore di due tentativi: il confronto fornisce evidenza per le attività testate, ma non garantisce risultati analoghi su tutti i carichi di lavoro o in prove ripetute.

  • Un risultato su un kernel di produzione

    Su MLA paged decode, un’esecuzione con budget aggiuntivo ha raggiunto uno speedup medio geometrico di 1.21x rispetto a FlashInfer in produzione, su 47 carichi di lavoro di produzione. La misura combina i rapporti tra la latenza della baseline e quella del kernel candidato; un valore superiore a 1 indica prestazioni aggregate più rapide.

    Il risultato mostra che, secondo questa misura, il kernel del team ha superato la baseline di produzione, pur potendo essere più lento su singoli carichi di lavoro.

  • I limiti delle evidenze

    Nell’analisi di MLA decode, un team fisso con quattro ruoli ha raggiunto uno speedup di 0.31x, contro 0.74x del team adattivo: il confronto suggerisce che l’adattamento può essere utile. Tuttavia, i passaggi del paper riportano versioni del modello incoerenti e non specificano budget di token equivalenti per questa analisi.

    Inoltre, nella migliore esecuzione con budget controllato su Gated Delta Net, 43 dei 100 carichi di lavoro erano più lenti della baseline, e ogni configurazione principale è stata eseguita solo due volte.

Dettagli

Paper successivoDynaFlow: imparare a coordinare agenti in modo flessibile