wasi-nn: Improve TPU support (#2447)

1. Allow TPU and GPU support at the same time. 2. Add Dockerfile to run example with [Coral USB](https://coral.ai/products/accelerator/).
2023-08-14 14:03:56 +02:00
parent 923e8d6471
commit b45d014112
10 changed files with 130 additions and 37 deletions
--- a/core/iwasm/libraries/wasi-nn/test/Dockerfile.nvidia-gpu
+++ b/core/iwasm/libraries/wasi-nn/test/Dockerfile.nvidia-gpu
@ -24,7 +24,7 @@ RUN apt-get install -y wget ca-certificates --no-install-recommends \

 RUN cmake \
    -DWAMR_BUILD_WASI_NN=1 \
-    -DWASI_NN_ENABLE_GPU=1 \
+    -DWAMR_BUILD_WASI_NN_ENABLE_GPU=1 \
    ..

 RUN make -j "$(grep -c ^processor /proc/cpuinfo)"