stoa opened a new pull request #7742:
URL: https://github.com/apache/tvm/pull/7742
We propose to contribute the STM32 port of the TVM.
The STM32 are mainly bare-metal boards and often use-cases do not suppose
using an OS: Linux, embed,RTOS,etc.
The STM32 development tools (ex, CubeMX) build standalone applications (no
OS) that can be flashed to the board and executed.
In this TVM port, the TVM is used to compile AI networks into a C code
compliant with the STM32 AI API.
In particular, the TVM is used to build a C module (kernels library + JSON
graph + params); then the stm32.emitter.py generates the C implementation from
the TVM module's JSON graph and writes out a C implementation of the kernels.
The generated C code can then be included as part of a larger STM32
application, and managed within a standard STM32 application development flow.
This contribution includes:
- a python/tvm/contrib/stm32 directory with scripts necessary for compiling
and communication with the STM32 board
- a src/runtime/stm32 directory that includes runtime support used by the
STM32 applications
- an apps/stm32 README and examples running in the STM32 DISCOVERY H747
board.
Tested on development board with several models including mnist, inceprion,
squeezenet, mobilenet, yolo, and a number of proprietary models.
This contribution does not modify any of the original TVM files.
@areusch
@tqchen
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
