diff --git a/core/iwasm/common/wasm_runtime_common.c b/core/iwasm/common/wasm_runtime_common.c index 685c7de04..51ecf23f9 100644 --- a/core/iwasm/common/wasm_runtime_common.c +++ b/core/iwasm/common/wasm_runtime_common.c @@ -1725,12 +1725,12 @@ wasi_nn_graph_registry_set_args(struct wasi_nn_graph_registry *registry, return true; } -int +static int wasi_nn_graph_registry_create(struct wasi_nn_graph_registry **registryp) { struct wasi_nn_graph_registry *args = wasm_runtime_malloc(sizeof(*args)); if (args == NULL) { - return false; + return -1; } wasm_runtime_wasi_nn_graph_registry_args_set_defaults(args); *registryp = args; diff --git a/core/iwasm/libraries/wasi-nn/src/wasi_nn.c b/core/iwasm/libraries/wasi-nn/src/wasi_nn.c index 8bb7e57cd..892ca5dd7 100644 --- a/core/iwasm/libraries/wasi-nn/src/wasi_nn.c +++ b/core/iwasm/libraries/wasi-nn/src/wasi_nn.c @@ -35,7 +35,7 @@ #define LLAMACPP_BACKEND_LIB "libwasi_nn_llamacpp" LIB_EXTENTION #define ONNX_BACKEND_LIB "libwasi_nn_onnx" LIB_EXTENTION -#define MAX_GLOBAL_GRAPHS_PER_INST 4 // ONNX only allows 4 graphs per instance +#define MAX_GLOBAL_GRAPHS_PER_INST 4 /* Global variables */ static korp_mutex wasi_nn_lock; diff --git a/core/iwasm/libraries/wasi-nn/test/requirements.txt b/core/iwasm/libraries/wasi-nn/test/requirements.txt index 0c80fd6b1..2145e3736 100644 --- a/core/iwasm/libraries/wasi-nn/test/requirements.txt +++ b/core/iwasm/libraries/wasi-nn/test/requirements.txt @@ -1,2 +1,2 @@ -tensorflow==2.14.0 +tensorflow==2.12.0 numpy==1.24.4