mirror of
https://github.com/bytecodealliance/wasm-micro-runtime.git
synced 2026-01-17 23:06:39 +00:00
Fix bugs
This commit is contained in:
parent
1cca0b8789
commit
9e039707c2
|
|
@ -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;
|
||||
|
|
|
|||
|
|
@ -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;
|
||||
|
|
|
|||
|
|
@ -1,2 +1,2 @@
|
|||
tensorflow==2.14.0
|
||||
tensorflow==2.12.0
|
||||
numpy==1.24.4
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user