gnh1201/wasm-micro-runtime
1
0
Fork 0
mirror of https://github.com/bytecodealliance/wasm-micro-runtime.git synced 2025-09-06 09:51:27 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

wasi_nn_llamacpp.c: explicitly reject unimplemented input index (#4446)

Browse Source 
note: wasmedge seems to accept index 1 for metadata. we don't
implement it.
This commit is contained in:
YAMAMOTO Takashi 2025-07-09 11:35:46 +09:00 committed by GitHub
parent d3b0b5c066
commit fbd27e5e03
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c
View File

@ -407,6 +407,11 @@ set_input(void *ctx, graph_execution_context exec_ctx, uint32_t index,
return runtime_error;
}
if (index != 0) {
NN_ERR_PRINTF("Invalid input index %d", index);
return invalid_argument;
}
// tensor->data is the prompt string.
char *prompt_text = (char *)wasi_nn_tensor->data.buf;
uint32_t prompt_text_len = wasi_nn_tensor->data.size;