christoph/wamr
1
Fork 0
Code Releases Activity

add load_by_name in wasi-nn (#4298)

Browse Source
This commit is contained in:
hongxia
2025-06-03 06:26:58 +08:00
committed by GitHub
parent 2a303861cc
commit aa1ff778b9
4 changed files with 26 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/iwasm/libraries/wasi-nn/test/utils.c
View File

@ -58,7 +58,7 @@ wasm_load(char *model_name, graph *g, execution_target target)
wasi_nn_error
wasm_load_by_name(const char *model_name, graph *g)
{
wasi_nn_error res = load_by_name(model_name, g);
wasi_nn_error res = load_by_name(model_name, strlen(model_name), g);
return res;
}
@ -108,7 +108,8 @@ run_inference(execution_target target, float *input, uint32_t *input_size,
uint32_t num_output_tensors)
{
graph graph;
if (wasm_load(model_name, &graph, target) != success) {
if (wasm_load_by_name(model_name, &graph) != success) {
NN_ERR_PRINTF("Error when loading model.");
exit(1);
}