Qualcomm® AI Engine Direct is designed to provide unified, low-level APIs for AI development. Qualcomm® AI Engine Direct is also referred to as QNN in the source and documentation. The QNN SDK can be downloaded from Qualcomm software center:
https://softwarecenter.qualcomm.com/#/catalog/item/a0844287-db23-11ed-a260-063166a9270b?type=Tool
Or from QPM [this option expected to be deprecated soon]
https://qpm.qualcomm.com/#/main/tools/details/qualcomm_ai_engine_direct
AI Engine Direct Helper(this repository) is also referred to as QAI AppBuilder in the source and documentation. QAI AppBuilder is extension for QNN SDK. We need some libraries in QNN SDK for using QAI AppBuilder.
QAI AppBuilder is designed for developer to using QNN SDK to execute model on Windows on Snapdragon(WoS) platforms easily. We encapsulated QNN SDK APIs to several simple APIs for loading the models to CPU or HTP and executing inference.
Developers can use QAI AppBuilder in both C++ and Python projects
• Support both C++ & Python
• Support multiple models.
• Support multiple inputs & outputs.
• Easier for developing apps.
• Faster for testing models.
Using the Python extensions with ARM64 Python will make it easier for developers to build GUI app for Windows on Snapdragon(WoS) platforms. Python 3.11.5 ARM64 version has support for following modules: PyQt6, OpenCV, Numpy, PyTorch*, Torchvision*, ONNX*. Developers can design apps that benefit from rich Python ecosystem.
*PyTorch, Torchvision, ONNX, ONNX Runtime: need to compile from source code. *Also support x64 Python.
There're two ways to use QAI AppBuilder:
Download prebuild binary package QAI_AppBuilder-win_arm64-{QNN SDK version}-Release.zip to get these files: https://github.com/quic/ai-engine-direct-helper/releases
libappbuilder.dll {libappbuilder.lib, LibAppBuilder.hpp} –– C++ projects can use this lib to run models in HTP. QAIAppSvc.exe –– Due to HTP limitations, we can only load models smaller than 4GB in one process. This app is used to help us load the models in new processes(Multiple processes can be created) and inference to avoid HTP restrictions.
Download Python extension qai_appbuilder-{version}-cp311-cp311-win_arm64.whl and install it with the command below: https://github.com/quic/ai-engine-direct-helper/releases
pip install qai_appbuilder-2.24.0-cp311-cp311-win_arm64.whl
Please refere to User Guide on how to use QAI AppBuilder in your project.
Build project with Visual Studio 2022 on WoS device:
- Set environment 'QNN_SDK_ROOT' to the QNN SDK path which you're using. E.g.: Set QNN_SDK_ROOT = "C:\Qualcomm\AIStack\QAIRT\2.24.0.240626\"
- Install Visual Studio 2022:
- Install Python-3.11.5 ARM64:
- Use the commands below to install Python dependency:
pip install wheel setuptools pybind11
- Download pybind11 repository to 'ai-engine-direct-helper\pybind\pybind11':
- Use the commands below to build and install Python extension(*.whl):
cd C:\Source\ai-engine-direct-helper
python setup.py bdist_wheel
# Install the extension:
pip install dist\qai_appbuilder-2.24.0-cp311-cp311-win_arm64.whl
QAI AppBuilder is licensed under the BSD 3-clause "New" or "Revised" License. Check out the LICENSE for more details.