christoph/wamr
1
Fork 0
Code Releases Activity

wasi-nn: Enable GPU support (#1922)

Browse Source 
- Split logic in several dockers
  - runtime: wasi-nn-cpu and wasi-nn- Nvidia-gpu.
  - compilation: wasi-nn-compile. Prepare the testing wasm and generates the TFLites.
- Implement GPU support for TFLite with Opencl.
This commit is contained in:
tonibofarull
2023-02-02 01:09:46 +01:00
committed by GitHub
parent fe3347d5d2
commit 1614ce12fa
10 changed files with 178 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
core/iwasm/libraries/wasi-nn/test/build.sh
View File

@ -7,7 +7,7 @@
-Wl,--allow-undefined \
-Wl,--strip-all,--no-entry \
--sysroot=/opt/wasi-sdk/share/wasi-sysroot \
-I/home/wamr/core/iwasm/libraries/wasi-nn \
-I.. \
-o test_tensorflow.wasm test_tensorflow.c
# TFLite models to use in the tests