Skip to content
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,6 +315,7 @@ def git_describe_version(original_version):
"micro_reference_vm.py",
"micro_tflite.py",
"micro_ethosu.py",
"micro_tvmc.py",
],
}

Expand Down
7 changes: 3 additions & 4 deletions gallery/how_to/work_with_microtvm/micro_autotune.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
"""
.. _tutorial-micro-autotune:

Autotuning with micro TVM
Autotuning with microTVM
=========================
**Authors**:
`Andrew Reusch <https://github.com/areusch>`_,
Expand All @@ -28,11 +28,10 @@
"""

import numpy as np
import subprocess
import pathlib

import tvm
from tvm.relay.backend import Executor, Runtime
from tvm.relay.backend import Runtime

####################
# Defining the model
Expand Down Expand Up @@ -67,7 +66,7 @@
params = {"weight": weight_sample}

#######################
# Defining the target #
# Defining the target
#######################
# Now we define the TVM target that describes the execution environment. This looks very similar
# to target definitions from other microTVM tutorials. Alongside this we pick the C Runtime to code
Expand Down
192 changes: 192 additions & 0 deletions gallery/how_to/work_with_microtvm/micro_tvmc.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,192 @@
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.

"""
.. _tutorial-micro-tvmc:

Executing a Tiny Model with TVMC Micro
=========================
**Author**: `Mehrdad Hessar <https://github.com/mehrdadh>`_

This tutorial explains how to compile a tiny model for a micro device,
build a program on Zephyr platform to execute this model, flash the program
and run the model all using `tvmc micro` command.
"""

######################################################################
# .. note::
# This tutorial is explaining using TVMC Mirco on Zephyr platform. You need
# to install Zephyr dependencies before processing with this tutorial. Alternatively,
# you can run this tutorial in one of the following ways which has Zephyr depencencies already installed.
#
# * Use `microTVM Reference Virtual Machines <https://tvm.apache.org/docs/how_to/work_with_microtvm/micro_reference_vm.html#sphx-glr-how-to-work-with-microtvm-micro-reference-vm-py>`_.
# * Use QEMU docker image provided by TVM. Following these you will download and login to the docker image:
# .. code-block:: bash
#
# cd tvm
# ./docker/bash.sh tlcpack/ci-qemu
#


############################################################
# Using TVMC Micro
############################################################
#
# TVMC is a python package which is installed as a part of TVM Python packages. Accessing this
# package varies based on your machine setup. First option is to use ``tvmc`` command directly.
# Alternatively, if you have TVM as a Python module on your ``$PYTHONPATH``, you can access this
# driver with ``python -m tvm.driver.tvm`` command. This tutorial will use TVMC command as
# ``tvmc`` for simplicity.
#
# To check if you have TVMC command installed on your machine, you can run:
#
# .. code-block:: bash
#
# tvmc --help
#
# To compile a model for microtvm we use ``tvmc compile`` subcommand. The output of this command
# is used in next steps with ``tvmc micro`` subcommands. You can check the availability of TVMC Micro using:
#
# .. code-block:: bash
#
# tvmc micro --help
#
# The main tasks that you can perform using ``tvmc micro`` are ``create``, ``build`` and ``flash``.
# To read about specific options under a givern subcommand, use
# ``tvmc micro <subcommand> --help``. We will use each subcommand in this tutorial.
#

############################################################
# Obtain a Tiny Model
############################################################
#
# For this tutorial, we will use Magic Wand model from tflite micro. Magin Wand is a
# Depthwise Convolution Layer model which recognizes gestures with an accelerometer.
#
# For this tutorial we will be using the model in tflite format.
#
# .. code-block:: bash
#
# wget https://github.com/tensorflow/tflite-micro/raw/main/tensorflow/lite/micro/examples/magic_wand/magic_wand.tflite
#

############################################################
# Compiling a TFLite model to a Model Library Format
############################################################
#
# Model Library Format (MLF) is an output format that TVM provides for micro targets. MLF is a tarball
# containing a file for each piece of the TVM compiler output which can be used on micro targets outside
# TVM environment. Read more about `Model Library Format <https://tvm.apache.org/docs//arch/model_library_format.html>`_.
#
# Here, we generate a MLF file for `qemu_x86` Zephyr board. To generate MLF output for the `magic_wand` tflite model:
#
# .. code-block:: bash
#
# tvmc compile magic_wand.tflite \
# --target='c -keys=cpu -link-params=0 -model=host' \
# --runtime=crt \
# --runtime-crt-system-lib 1 \
# --executor='graph' \
# --executor-graph-link-params 0 \
# --output model.tar \
# --output-format mlf \
# --pass-config tir.disable_vectorize=1 \
# --disabled-pass=AlterOpLayout
#
# This will generate a `model.tar` file which contains TVM compiler output files. To run this command for
# a different Zephyr device, you need to update `target`. For instance, for `nrf5340dk_nrf5340_cpuapp` board
# the target is ``--target='c -keys=cpu -link-params=0 -model=nrf5340dk'``.
#


############################################################
# Create a Zephyr Project Using Model Library Format
############################################################
#
# To generate a Zephyr project we use TVM Micro subcommand ``create``. We pass the MLF format and the path
# for the project to ``create`` subcommand along with project options. Project options for each
# platform (Zephyr/Arduino) are defined in their Project API server file. To generate Zephyr project, run:
#
# .. code-block:: bash
#
# tvmc micro create \
# project \
# model.tar \
# zephyr \
# --project-option project_type=host_driven zephyr_board=qemu_x86
#
# This will generate a Zephyr project for `qemu_x86` zephyr board. To get more information about
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

could you explain host_driven here?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added some details, PTAL. thanks!

# TVMC Micro ``create`` subcommand:
#
# .. code-block:: bash
#
# tvmc micro create --help
#

############################################################
# Build and Flash Zephyr Project Using TVMC Micro
############################################################
#
# Next step is to build the Zephyr project which includes TVM generated code for running the tiny model, Zephyr
# template code to run a model in host_driven mode and TVM runtime source/header files. To build the project:
#
# .. code-block:: bash
#
# tvmc micro build \
# project \
# zephyr \
# --project-option zephyr_board=qemu_x86
#
# This will build the project in `project` directory and generates binary files under `project/build`. To build
# Zephyr project for a different Zephyr board, change `zephyr_board` project option.
#
# Next, we flash the Zephyr binary file to Zephyr device. For `qemu_x86` Zephyr board this step does not
# actually perform any action since QEMU will be used, however you need this step for physical hardware.
#
# .. code-block:: bash
#
# tvmc micro flash \
# project \
# zephyr \
# --project-option zephyr_board=qemu_x86
#

############################################################
# Run Tiny Model on Micro Target
############################################################
#
# To run the flashed model on the device using TVMC, we use ``tvmc run`` subcommand and we pass ``--device micro``
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

could you explain what exactly has happened now at this point, and what we mean by "run the flashed model on the device?"

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@areusch added more details. Please let me know if I should mention anything else. thanks!

# to specify the device type.
#
# .. code-block:: bash
#
# tvmc run \
# --device micro \
# project \
# --project-option zephyr_board=qemu_x86 \
# --fill-mode ones
# --print-top 4
# # Output:
# #
# # INFO:__main__:b'[100%] [QEMU] CPU: qemu32,+nx,+pae\n'
# # remote: microTVM Zephyr runtime - running
# # INFO:__main__:b'[100%] Built target run\n'
# # [[3. 1. 2. 0. ]
# # [0.47213247 0.41364592 0.07525456 0.03896701]]
#
# This command sets the input of the model to all ones, runs the model on Zephyr board and shows the top
# four values of the output with their indices.
2 changes: 1 addition & 1 deletion tests/micro/common/test_tvmc.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ def test_tvmc_model_build_only(board, output_dir):
shutil.rmtree(out_dir_temp)
os.mkdir(out_dir_temp)

model_path = model_path = download_testdata(MODEL_URL, MODEL_FILE, module="data")
model_path = download_testdata(MODEL_URL, MODEL_FILE, module="data")
tar_path = str(output_dir / "model.tar")
project_dir = str(output_dir / "project")

Expand Down