From fbd27e5e0345f86e760a7169d0b6e5ca7cce3635 Mon Sep 17 00:00:00 2001 From: YAMAMOTO Takashi Date: Wed, 9 Jul 2025 11:35:46 +0900 Subject: [PATCH] wasi_nn_llamacpp.c: explicitly reject unimplemented input index (#4446) note: wasmedge seems to accept index 1 for metadata. we don't implement it. --- core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c b/core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c index 7f30f2b4d..5e985ec35 100644 --- a/core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c +++ b/core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c @@ -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;