Просмотр исходного кода

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

note: wasmedge seems to accept index 1 for metadata. we don't
implement it.
YAMAMOTO Takashi 6 месяцев назад
Родитель
Сommit
fbd27e5e03
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c

+ 5 - 0
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;