El driver NPU Rockchip “Rocket”, desarrollado por Tomeu Vizoso, está a punto de integrarse en el kernel Linux 6.18, ofreciendo soporte open-source para los aceleradores de inteligencia artificial de Rockchip.
Este avance permitirá a los usuarios de Linux aprovechar la potencia de los NPUs de Rockchip directamente desde el kernel, sin depender de blobs propietarios.
Rocket: Driver NPU Open Source para Linux
El driver Rocket es un esfuerzo de ingeniería inversa que ha estado en desarrollo por más de año y medio.
Actualmente, el código del kernel está en la rama DRM-Misc-Next, y se espera que llegue al kernel principal Linux 6.18, aunque aún podría considerarse para v6.17 si hay cambios de última hora.
Este driver permite la aceleración de tareas de AI directamente en el hardware NPU de Rockchip, facilitando la ejecución de modelos de inferencia y clasificación de imágenes en dispositivos como SBCs, laptops y sistemas embebidos basados en Rockchip.
Mesa 25.3 y Rocket Gallium3D
Paralelamente, el driver Rocket para Gallium3D fue integrado en Mesa 25.3, habilitando soporte en user-space para programar los NPUs desde aplicaciones y frameworks de AI.
Este driver utiliza el framework Gallium3D TEFLON, con un modelo de programación similar al de NVIDIA NVDLA.
Actualmente, el driver permite ejecutar modelos como SSDLite MobileDet con un rendimiento comparable al del firmware propietario en un solo núcleo NPU, marcando un paso importante hacia un ecosistema completamente open-source en aceleración AI.
Futuro del driver Rocket
Tomeu Vizoso planea extender el soporte a otros SoCs de Rockchip, optimizar el rendimiento y mejorar la integración con otros drivers de NPU, incluyendo Etnaviv para los chips Vivante.
Esto abrirá nuevas posibilidades para desarrolladores, investigadores y fabricantes que quieran AI en Linux sin depender de controladores privativos.
Conclusión
Con la llegada de Linux 6.18 y Mesa 25.3, los NPUs de Rockchip ganan finalmente soporte open-source robusto.
Esto representa un gran paso para inteligencia artificial en Linux, democratizando el acceso a aceleradores AI y permitiendo entornos totalmente libres de software propietario.