Skip to content
forked from Xilinx/mlir-aie

An MLIR-based toolchain for Xilinx Versal AIEngine-based devices.

License

Notifications You must be signed in to change notification settings

david-vc/mlir-aie

 
 

Repository files navigation

MLIR-based AI Engine toolchain

GitHub Workflow Status GitHub Pull Requests

This repository contains an MLIR-based toolchain for AI Engine-enabled devices, such as AMD Ryzen™ AI and Versal™. This repository can be used to generate low-level configurations for the AI Engine portion of these devices. AI Engines are organized as a spatial array of tiles, where each tile contains AI Engine cores and/or memories. The spatial array is connected by stream switches that can be configured to route data between AI Engine tiles scheduled by their programmable Data Movement Accelerators (DMAs). This repository contains MLIR representations, with multiple levels of abstraction, to target AI Engine devices. This enables compilers and developers to program AI Engine cores, as well as describe data movements and array connectivity. A Python API is made available as a convenient interface for generating MLIR design descriptions. Backend code generation is also included, targeting the aie-rt library. This toolchain uses the AI Engine compiler tool which is part of the AMD Vitis™ software installation: these tools require a free license for use from the Product Licensing Site.

This project is primarily intended to support the open-source community, particularly tool builders, with low-level access to AIE devices and enable the development of a wide variety of programming models from higher level abstractions. As such, although it contains some examples, this project is not intended to represent an end-to-end compilation flow for application design. If you're looking for an out-of-the-box experience for highly efficient machine learning, check out the AMD Ryzen™ AI Software Platform.

Getting Started

Running on a Versal™ board

Getting Started and Running on Windows Ryzen™ AI

Full Documentation


Copyright© 2019-2023 Advanced Micro Devices, Inc

About

An MLIR-based toolchain for Xilinx Versal AIEngine-based devices.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • MLIR 59.7%
  • C++ 29.6%
  • Python 5.5%
  • Tcl 2.1%
  • CMake 1.1%
  • C 0.9%
  • Other 1.1%