From 521e281ead3bed140135f417458bcdad88a1b436 Mon Sep 17 00:00:00 2001 From: Ahmedounet Date: Mon, 13 Jun 2022 12:50:44 +0200 Subject: [PATCH] fix: "compilation bug fixed" --- build-scripts/runtime_lib.cmake | 3 ++- core/iwasm/libraries/wasi-nn/wasi_nn.c | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/build-scripts/runtime_lib.cmake b/build-scripts/runtime_lib.cmake index 461a64b37..3127fd66b 100644 --- a/build-scripts/runtime_lib.cmake +++ b/build-scripts/runtime_lib.cmake @@ -81,7 +81,8 @@ if (WAMR_BUILD_WASI_NN EQUAL 1) RESULT_VARIABLE TENSORFLOW_RESULT ) set(TENSORFLOW_SOURCE_DIR "/root/src/tflite/tensorflow_src") - + include_directories (${CMAKE_CURRENT_BINARY_DIR}/flatbuffers/include) + include_directories (${TENSORFLOW_SOURCE_DIR}) add_subdirectory( "${TENSORFLOW_SOURCE_DIR}/tensorflow/lite" "${CMAKE_CURRENT_BINARY_DIR}/tensorflow-lite" EXCLUDE_FROM_ALL) diff --git a/core/iwasm/libraries/wasi-nn/wasi_nn.c b/core/iwasm/libraries/wasi-nn/wasi_nn.c index 567080f67..422c9316c 100644 --- a/core/iwasm/libraries/wasi-nn/wasi_nn.c +++ b/core/iwasm/libraries/wasi-nn/wasi_nn.c @@ -25,7 +25,6 @@ uint32_t wasi_nn_load(wasm_exec_env_t exec_env, uint32_t builder, uint32_t encod printf("Inside wasi_nn_load!\n\n"); wasm_module_inst_t instance = wasm_runtime_get_module_inst(exec_env); graph_builder_array buf = (graph_builder_array) wasm_runtime_addr_app_to_native(instance, builder); - load_return res; switch ((graph_encoding) encoding) { case openvino: return invalid_argument;