From ac9bb989911ff86bd850d27d2afcb1d71f2325c6 Mon Sep 17 00:00:00 2001
From: "Rob Herring (Arm)" <robh@kernel.org>
Date: Thu, 19 Mar 2026 12:08:13 -0500
Subject: [PATCH] test_teflon: Fix missing UInt16/Int16 output size

The output comparison expects number of elements, not bytes. Adjusting the
size was missed when adding UInt16/Int16 output comparisons.

Signed-off-by: Rob Herring (Arm) <robh@kernel.org>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/40525>
---
 src/gallium/targets/teflon/test_executor.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/gallium/targets/teflon/test_executor.cpp b/src/gallium/targets/teflon/test_executor.cpp
index 4683d7a671e..f0035694f4f 100644
--- a/src/gallium/targets/teflon/test_executor.cpp
+++ b/src/gallium/targets/teflon/test_executor.cpp
@@ -526,6 +526,11 @@ run_model(TfLiteModel *model, enum executor executor, void ***input, size_t *num
          (*output_sizes)[i] = output_tensor->bytes / 4;
          break;
       }
+      case kTfLiteInt16:
+      case kTfLiteUInt16: {
+         (*output_sizes)[i] = output_tensor->bytes / 2;
+         break;
+      }
       default: {
          (*output_sizes)[i] = output_tensor->bytes;
          break;