Vitis

Transformer 模型针对 AMD AI 引擎上的 CERN 喷注标记进行了优化

研究人员开发了一种方法，将 transformer 模型部署到 AMD Versal AI 引擎上进行喷注标记，这项任务对于 CERN 大型强子对撞机的触发系统至关重要。该方法涉及一个量化的、仅整数的 transformer，它将密集和多头注意力层映射到 AI 引擎块。一项关键贡献是一个可重用的软件框架，可以从 Python 模型描述生成 Vitis 图代码，从而支持未来的研究并提供开源解决方案。