HardenedLinux社区发布AI运行时框架SonicBoom 以图编译器优化实现推理性能提升

研究和生产部署使用的是完全相同的运行时图编译器优化栈,这意味着研发完就能直接部署到产品。

HardenedLinux社区主导开发的全新AI运行时框架SonicBoom正式发布,采用GPLv3许可证。该框架提供语言无关的C API接口,开发者可通过任意语言绑定该库,从而在不同编程语言环境中进行高性能AI研发。目前GNU Guile为官方支持语言,并利用Guile的delimited continuation在应用层实现“死计算消除”,仅有效计算进入运行时。

在运行时层面,SonicBoom集成LLVM MLIR,对计算图进行二次优化以进一步剔除无效计算图,并借助LLVM根据硬件具体情况生成kernel,替代传统的手工硬件调优方式。这一设计使应用层绑定的动态语言兼具PyTorch的灵活性,同时在单点推理性能上超越llama.cpp。框架在运行时自动对计算图进行编译优化,同时实现PyTorch的动态图弹性与TensorFlow的静态图性能。

针对推理部署场景,SonicBoom无需应用层动态语言支持,可直接加载模型进行高效推理。研究与生产部署使用完全相同的运行时图编译器优化栈,研发完成后可直接部署至产品环境。部署成本被压缩至一个三百多MB的.so文件,无需数十GB的Python依赖环境。

性能方面,SonicBoom直接复用llama.cpp已有的kernel,在kernel优化不变的前提下依靠图编译器优化,在RTX 3050 8G平台上实现原始性能提升20%,且优化仍在持续。llama.cpp此前依靠经验丰富的工程师进行kernel调优,已将性能推至较高水平,SonicBoom则从图编译层面寻找新的性能空间。

目前SonicBoom社区版仅提供单点推理能力,分布式推理能力计划在企业版中提供。该框架的对标目标为PyTorch,后续计划将更高性能扩展至image、video、TTS等计算场景。硬件支持上,当前仅支持CPI和CUDA,之后计划支持Vulkan,为non-NVIDIA硬件及纯自由软件爱好者提供选择。

由于缺乏多样化的GPU评测数据,SonicBoom在RTX 3050之外的平台测试数据可能存在波动。v0.0.3版本正在邀请社区贡献者参与不同NVIDIA平台的评测,社区提供了简化脚本以降低参与门槛,每位测评贡献者将列入贡献者名单。

从行业角度看,SonicBoom试图以图编译器优化替代手工kernel调优的路径,为AI推理框架提供了不同于现有方案的技术思路。其语言无关的C API设计与统一的研究生产优化栈,若能按计划扩展至多模态计算与企业级分布式推理,或将对AI基础设施层的工具链格局产生一定影响。

该文观点仅代表作者本人,企服科学平台仅提供信息存储空间服务。

赞 (0)
研究显示生成式AI可植入虚假记忆,MIT实验揭示认知风险
上一篇 4小时前
万兴科技发布UniConverter 18,AI媒体处理效率最高提升约90%
下一篇 2小时前

相关推荐

发表回复

登录后才能评论
分享本页
返回顶部