mirror of
https://github.com/bytecodealliance/wasm-micro-runtime.git
synced 2025-02-08 16:05:07 +00:00
1614ce12fa
- 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.
21 lines
499 B
Bash
Executable File
21 lines
499 B
Bash
Executable File
# Copyright (C) 2019 Intel Corporation. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
# WASM application that uses WASI-NN
|
|
|
|
/opt/wasi-sdk/bin/clang \
|
|
-Wl,--allow-undefined \
|
|
-Wl,--strip-all,--no-entry \
|
|
--sysroot=/opt/wasi-sdk/share/wasi-sysroot \
|
|
-I.. \
|
|
-o test_tensorflow.wasm test_tensorflow.c
|
|
|
|
# TFLite models to use in the tests
|
|
|
|
cd models
|
|
python3 average.py
|
|
python3 max.py
|
|
python3 mult_dimension.py
|
|
python3 mult_outputs.py
|
|
python3 sum.py
|