christoph/wamr
1
Fork 0
Code Releases Activity

sync up with latest wasi-nn spec (#3530)

Browse Source
This commit is contained in:
liang.he
2024-06-17 14:58:09 +08:00
committed by GitHub
parent d3e89895be
commit db025e457a
9 changed files with 209 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23,14 +23,14 @@ wasm_load(char *model_name, graph *g, execution_target target)
buffer = (uint8_t *)malloc(sizeof(uint8_t) * MAX_MODEL_SIZE);
if (buffer == NULL) {
fclose(pFile);
return missing_memory;
return too_large;
}
result = fread(buffer, 1, MAX_MODEL_SIZE, pFile);
if (result <= 0) {
fclose(pFile);
free(buffer);
return missing_memory;
return too_large;
}
graph_builder_array arr;
@ -40,7 +40,7 @@ wasm_load(char *model_name, graph *g, execution_target target)
if (arr.buf == NULL) {
fclose(pFile);
free(buffer);
return missing_memory;
return too_large;
}
arr.buf[0].size = result;
@ -54,6 +54,13 @@ wasm_load(char *model_name, graph *g, execution_target target)
return res;
}
wasi_nn_error
wasm_load_by_name(const char *model_name, graph *g)
{
wasm_nn_error res = load_by_name(model_name, g);
return res;
}
wasi_nn_error
wasm_init_execution_context(graph g, graph_execution_context *ctx)
{
@ -67,7 +74,7 @@ wasm_set_input(graph_execution_context ctx, float *input_tensor, uint32_t *dim)
dims.size = INPUT_TENSOR_DIMS;
dims.buf = (uint32_t *)malloc(dims.size * sizeof(uint32_t));
if (dims.buf == NULL)
return missing_memory;
return too_large;
tensor tensor;
tensor.dimensions = &dims;