基于MLIR的AI引擎工具链
该仓库包含一个基于MLIR的工具链,用于支持AI引擎的设备,如AMD Ryzen™ AI和Versal™。这个仓库可用于为这些设备的AI引擎部分生成低级配置。AI引擎被组织为一个空间阵列的瓦片,每个瓦片包含AI引擎核心和/或内存。这个空间阵列通过可配置的流交换机连接,可以路由数据在AI引擎瓦片之间,由其可编程的数据移动加速器(DMAs)进行调度。该仓库包含多个抽象层次的MLIR表示,以针对AI引擎设备。这使得编译器和开发人员能够编程AI引擎核心,以及描述数据移动和阵列连接。我们提供了一个Python API作为生成MLIR设计描述的便捷接口。后端代码生成也包含在内,目标是aie-rt库。这个工具链使用AMD Vitis™软件安装中的AI引擎编译器工具:这些工具需要从产品许可网站获得免费许可证才能使用。
这个项目主要旨在支持开源社区,特别是工具构建者,提供对AIE设备的低级访问,并支持从更高级别抽象开发各种编程模型。我们提供了一个示例编程流程:用于实践的接口表示(IRON),用于AIE阵列的接近底层的编程。IRON是一个开放访问工具包,使性能工程师能够通过围绕mlir-aie方言的一组Python语言绑定,构建快速高效,通常是专门化的设计。因此,它包含一些示例,但这个项目并不打算代表所有应用设计的端到端编译流程。如果你正在寻找高效机器学习的开箱即用体验,请查看AMD Ryzen™ AI软件平台。
版权所有© 2019-2024 Advanced Micro Devices, Inc