commit 758e73e117c5cef17f8f0b1c543efc1df953b2fa
author Jerry Ge <jerry.ge@arm.com> Mon Feb 26 13:31:22 2024 -0800
committer Jerry Ge <jerry.ge@arm.com> Tue Feb 27 10:35:07 2024 -0800
tree 2033377d30b702abff184884bd87daeba90d2d91
parent 81db8ee8f580d30ec0ca53067df32ef046e6f09e

Update ConvertU8toF16 to save outputs in F16 containers

- If we save FP16 in FP32 containers, it will change byte representation
- which causes wrong values after deserialization
- Update the code to store F16 values in F16 containers

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I2ddea67471997c596980faea1419422d96bad60d

include/tosa_serialization_handler.h

diff --git a/include/tosa_serialization_handler.h b/include/tosa_serialization_handler.h
index f50d7ad..5c53f57 100644
--- a/include/tosa_serialization_handler.h
+++ b/include/tosa_serialization_handler.h
@@ -421,7 +421,8 @@
     static tosa_err_t ConvertI4toU8(const std::vector<int8_t>& in, std::vector<uint8_t>& out);
     static tosa_err_t ConvertBooltoU8(const std::vector<bool>& in, std::vector<uint8_t>& out);
 
-    static tosa_err_t ConvertU8toF16(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<float>& out);
+    static tosa_err_t
+        ConvertU8toF16(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<half_float::half>& out);
     static tosa_err_t ConvertU8toF32(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<float>& out);
     static tosa_err_t ConvertU8toI64(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int64_t>& out);
     static tosa_err_t ConvertU8toI48(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int64_t>& out);