From db7714f0f5614b567fd6d521191adffcf03b2f59 Mon Sep 17 00:00:00 2001 From: YAMAMOTO Takashi Date: Wed, 18 Jun 2025 20:08:57 +0900 Subject: [PATCH] wasi_nn_tensorflowlite.cpp: reject non-fp32 input earlier (#4388) this backend assumes fp32 here and there. it's safer to reject unexpected inputs explicitly. --- core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.cpp b/core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.cpp index 09e12f0d2..9c458179a 100644 --- a/core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.cpp +++ b/core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.cpp @@ -281,6 +281,11 @@ set_input(void *tflite_ctx, graph_execution_context ctx, uint32_t index, { TFLiteContext *tfl_ctx = (TFLiteContext *)tflite_ctx; + if (input_tensor->type != fp32) { + NN_ERR_PRINTF("unsupported input tensor type %u", input_tensor->type); + return runtime_error; + } + wasi_nn_error res; if (success != (res = is_valid_graph_execution_context(tfl_ctx, ctx))) return res;