dsg/mesa
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test_teflon: Fix missing UInt16/Int16 output size

Browse source 
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>
This commit is contained in:
Rob Herring (Arm) 2026-03-19 12:08:13 -05:00 • committed by Marge Bot
parent 6375db3fd3
commit ac9bb98991
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/gallium/targets/teflon/test_executor.cpp
View file

@ -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;