mirror of
https://github.com/bytecodealliance/wasm-micro-runtime.git
synced 2025-05-10 05:36:15 +00:00
ed6b8efade
Since the Tensorflow library is already installed in many cases(especially in the case of the embedded system), move the installation code to find_package.
23 lines
622 B
CMake
23 lines
622 B
CMake
# Copyright (C) 2019 Intel Corporation. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_LIST_DIR}/cmake)
|
|
|
|
# Find tensorflow-lite
|
|
find_package(tensorflow_lite REQUIRED)
|
|
|
|
set (WASI_NN_DIR ${CMAKE_CURRENT_LIST_DIR})
|
|
|
|
include_directories (${WASI_NN_DIR})
|
|
include_directories (${WASI_NN_DIR}/src)
|
|
include_directories (${WASI_NN_DIR}/src/utils)
|
|
|
|
set (
|
|
LIBC_WASI_NN_SOURCE
|
|
${WASI_NN_DIR}/src/wasi_nn.c
|
|
${WASI_NN_DIR}/src/wasi_nn_tensorflowlite.cpp
|
|
${WASI_NN_DIR}/src/utils/wasi_nn_app_native.c
|
|
)
|
|
|
|
set (TENSORFLOW_LIB tensorflow-lite)
|