gnh1201/wasm-micro-runtime
1
0
Fork 0
mirror of https://github.com/bytecodealliance/wasm-micro-runtime.git synced 2025-02-08 16:05:07 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity
0435acdd43
wasm-micro-runtime/core/iwasm/libraries/wasi-nn/README.md
tonibofarull 9eed6686df
Refactor WASI-NN to simplify the support for multiple frameworks (#1834) 
- Reorganize the library structure
- Use the latest version of `wasi-nn` wit (Oct 25, 2022):
    0f77c48ec1/wasi-nn.wit.md
- Split logic that converts WASM structs to native structs in a separate file
- Simplify addition of new frameworks
2023-01-25 18:32:40 +08:00

966 B
Raw Blame History

WASI-NN

How to use

Enable WASI-NN in the WAMR by spefiying it in the cmake building configuration as follows,

set (WAMR_BUILD_WASI_NN  1)

The definition of the functions provided by WASI-NN is in the header file core/iwasm/libraries/wasi-nn/wasi_nn.h.

By only including this file in your WASM application you will bind WASI-NN into your module.

Tests

To run the tests we assume that the current directory is the root of the repository.

  1. Build the docker image,
docker build -t wasi-nn -f core/iwasm/libraries/wasi-nn/test/Dockerfile .
  1. Run the container
docker run wasi-nn

If all the tests have run properly you will the the following message in the terminal,

Tests: passed!

What is missing

Supported:

  • Only 1 WASM app at a time.
  • Only 1 model at a time.
    • graph and graph-execution-context are ignored.
  • Graph encoding: tensorflowlite.
  • Execution target: cpu.
  • Tensor type: fp32.