Oossa

llama.cpp accélère Zamba2 sur GPU jusqu’à 15,1 tokens/s

La mise à jour de ggml-cuda ajoute une voie rapide pour d_state=64 : la génération de Zamba2‑7B‑Instruct passe de 11 à 15,1 tokens par seconde.

BrèvePar Publié par Oossa: 1 min de lecture

Le projet llama.cpp a publié une modification du noyau GPU qui ajoute une voie rapide pour une taille d’état de 64. Zamba2 et des puces similaires utilisent cette taille ; avant ce changement, chaque étape du scan était exécutée sur le CPU, ce qui limitait la génération à environ 6‑7 tokens par seconde. Le nouveau modèle de calcul augmente le débit d’environ 35 % : sur le même matériel, Zamba2‑7B‑Instruct passe de 11 à 15,1 tokens par seconde. Certaines opérations de scan (environ 28 %) sont toutefois toujours exécutées sur le CPU : la parité complète n’est donc pas encore atteinte.

Pourquoi c'est important

Les développeurs qui utilisent des GPU Zamba2 peuvent générer du texte nettement plus vite sans matériel supplémentaire.

Cet article vous a-t-il été utile ?
Partager

À lire ensuite

Oossa · Newsletter

La semaine de l'IA, expliquée

Chaque lundi : l'essentiel de l'actualité, en langage clair. Gratuit, sans spam.