Google Git
Sign in
opensecura / sw / vec_iree / cedef3ddb88d43209e978fe0618760206d817861
commitcedef3ddb88d43209e978fe0618760206d817861[log] [tgz]
authorLun Dong <lundong@google.com>Fri Jul 29 21:40:17 2022 -0700
committerLun Dong <lundong@google.com>Fri Jul 29 21:40:17 2022 -0700
tree10c142b5ac6ddeb782370d34f9c68feea2f87113
parentc4839b36b96d627acba585ff889c3901aa30b7bc [diff]
Bug fix on resizing input image

The input tuple for resizing an image is (width, height). It should
correspnds to (input_shape[2], input_shape[1]).

In mobilenet/mnist, input width = height, so this issue was not captured
before.

Change-Id: I6cf61bca3e8e2f5c163c8e5dec414cf674105324
1 file changed
tree: 10c142b5ac6ddeb782370d34f9c68feea2f87113
  1. build_tools/
  2. cmake/
  3. native_log/
  4. samples/
  5. CMakeLists.txt
  6. README.md
  7. springbok_config.h
README.md

Springbok Model Execution Repository

This project contains ML model codegen and execution libraries/examples. The artifacts built from this project is targeted for RISC-V 32-bit baremetal machine (Springbok).

Prerequisite

If you get this project from Project Shodan manifest, you are all set. If not, you need to have following projects as well to build the project successfully.

This project needs to be at <dir>/sw/vec_iree

Code structure

  • build_tools: Utility scripts for the project
  • cmake: CMake Macros for the project
  • native_log: BSP support for the Linux userspace emulation
  • samples: Codegen and execution of ML models based on IREE
    • audio_prep: Audio preprocessing libarary
    • device: Device HAL driver library
    • risp4ml: Vision preprocessing library (Reduced ISP for ML)
    • simple_vec_mul: Point-wise vector multiplication examples
    • util: Runtime utility library for model execution

Build the project

If you are running from Project Shodan repo, you can build the artifacts with m iree. If you download this project alone, you need to download/build IREE's host compiler, then at the top level directory

  cmake -B <output dir> -G Ninja \
     -DCMAKE_TOOLCHAIN_FILE="$(realpath sw/vec_iree/cmake/riscv_iree.cmake)" \
     -DCMAKE_BUILD_TYPE=MinSizeRel \
     -DIREE_HOST_BINARY_ROOT="$(IREE_COMPILER_DIR)/install" \
     -DRISCV_TOOLCHAIN_ROOT=$(TOOLCHAINRV32_PATH) \
     <dir>/sw/vec_iree
  cmake --build <output dir> --target all

Run the executables

You can run QEMU or RENODE emulation with the built executables. In Shodan repo, qemu_sim_springbok for QEMU emulation, and sim_springbok for Renode simulation. Out-of-repo emulation is possible but require more steps.

Test the executables

In Shodan, this project utilizes LLVM lit and FileCheck to test the ML executable performance. The tests are defined in the *_test.run files under <output dir>. To run the test, at the Shodan top level directory

lit --path <Filecheck dir> -a <output dir>

Filecheck in Debian testing is under /usr/lib/llvm-11/bin

Add -D RUNNER=qemu to enable the qemu tests.