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

4
core/iwasm/libraries/wasi-nn/src/utils/wasi_nn_app_native.c
View File

@ -76,7 +76,7 @@ graph_builder_array_app_native(wasm_module_inst_t instance,
graph_builder *builder = (graph_builder *)wasm_runtime_malloc(
array_size * sizeof(graph_builder));
if (builder == NULL)
return missing_memory;
return too_large;
for (uint32_t i = 0; i < array_size; ++i) {
wasi_nn_error res;
@ -149,7 +149,7 @@ tensor_dimensions_app_native(wasm_module_inst_t instance,
*dimensions =
(tensor_dimensions *)wasm_runtime_malloc(sizeof(tensor_dimensions));
if (dimensions == NULL)
return missing_memory;
return too_large;
(*dimensions)->size = dimensions_wasm->size;
(*dimensions)->buf = (uint32_t *)wasm_runtime_addr_app_to_native(