Vitis-AI Execution Provider

Vitis-AI is Xilinx’s development stack for hardware-accelerated AI inference on Xilinx platforms, including both edge devices and Alveo cards. It consists of optimized IP, tools, libraries, models, and example designs. It is designed with high efficiency and ease of use in mind, unleashing the full potential of AI acceleration on Xilinx FPGA and ACAP.

The current Vitis-AI execution provider inside ONNXRuntime enables acceleration of Neural Network model inference using DPUv1. DPUv1 is a hardware accelerator for Convolutional Neural Networks (CNN) on top of the Xilinx Alveo platform and targets U200 and U250 accelerator cards.

Contents

• Requirements
• Build
• Usage
• Configuration Options
• Samples

Requirements

The following table lists system requirements for running docker containers as well as Alveo cards.

Build

See Build instructions.

Hardware setup

1. Clone the Vitis AI repository:

    git clone https://github.com/xilinx/vitis-ai
   

2. Install the Docker, and add the user to the docker group. Link the user to docker installation instructions from the following docker’s website:
   • https://docs.docker.com/install/linux/docker-ce/ubuntu/
   • https://docs.docker.com/install/linux/docker-ce/centos/
   • https://docs.docker.com/install/linux/linux-postinstall/
3. Any GPU instructions will have to be separated from Vitis AI.
4. Set up Vitis AI to target Alveo cards. To target Alveo cards with Vitis AI for machine learning workloads, you must install the following software components:
   • Xilinx Runtime (XRT)
   • Alveo Deployment Shells (DSAs)
   • Xilinx Resource Manager (XRM) (xbutler)
   • Xilinx Overlaybins (Accelerators to Dynamically Load - binary programming files)

   While it is possible to install all of these software components individually, a script has been provided to automatically install them at once. To do so:

   • Run the following commands:

     cd Vitis-AI/alveo/packages
     sudo su
     ./install.sh
     

   • Power cycle the system.
5. Build and start the ONNXRuntime Vitis-AI Docker Container.

   cd {onnxruntime-root}/dockerfiles
   docker build -t onnxruntime-vitisai -f Dockerfile.vitisai .
   ./scripts/docker_run_vitisai.sh
   

   Setup inside container

   source /opt/xilinx/xrt/setup.sh
   conda activate vitis-ai-tensorflow
   

Usage

On-the-fly quantization

Usually, to be able to accelerate inference of Neural Network models with Vitis-AI DPU accelerators, those models need to quantized upfront. In the ONNXRuntime Vitis-AI execution provider we make use of on-the-fly quantization to remove this additional preprocessing step. In this flow, one doesn’t need to quantize his/her model upfront but can make use of the typical inference execution calls (InferenceSession.run) to quantize the model on-the-fly using the first N inputs that are provided (see more information below). This will set up and calibrate the Vitis-AI DPU and from that point onwards inference will be accelerated for all next inputs.

Configuration Options

A couple of environment variables can be used to customize the Vitis-AI execution provider.

Samples

When using python, you can base yourself on the following example:

# Import pyxir before onnxruntime
import pyxir
import pyxir.frontend.onnx
import pyxir.contrib.dpuv1.dpuv1

import onnxruntime

# Add other imports 
# ...

# Load inputs and do preprocessing
# ...

# Create an inference session using the Vitis-AI execution provider
session = onnxruntime.InferenceSession('[model_file].onnx', None,["VitisAIExecutionProvider"])

# First N (default = 128) inputs are used for quantization calibration and will
#   be executed on the CPU
# This config can be changed by setting the 'PX_QUANT_SIZE' (e.g. export PX_QUANT_SIZE=64)
imput_name = [...]
outputs = [session.run([], {input_name: calib_inputs[i]})[0] for i in range(128)]

# Afterwards, computations will be accelerated on the FPGA
input_data = [...]
result = session.run([], {input_name: input_data})