From c5291695f04901e8abbc26dad6cba10e2c7685f8 Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Tue, 2 Jan 2024 22:29:08 +0000
Subject: Save Int8/UInt8 reference outputs to native dtypes

* Int8/UInt8 reference outputs were previously saved to INT32
* Save those in their native dtypes and updated other affected code

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Georgios Pinitas <georgios.pinitas@arm.com>

Change-Id: Id65fc8773150d3c56bc2c72789a6a0d3c78cd363
---
 reference_model/src/model_runner.cc | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'reference_model/src/model_runner.cc')

diff --git a/reference_model/src/model_runner.cc b/reference_model/src/model_runner.cc
index 6f65202..28ad72c 100644
--- a/reference_model/src/model_runner.cc
+++ b/reference_model/src/model_runner.cc
@@ -82,12 +82,16 @@ int IModelRunner::getOutput(std::string output_name, uint8_t* raw_ptr, size_t si
 // Template explicit specialization
 template int IModelRunner::setInput<float>(std::string input_name, std::vector<float>& vals);
 template int IModelRunner::setInput<half_float::half>(std::string input_name, std::vector<half_float::half>& vals);
+template int IModelRunner::setInput<int8_t>(std::string input_name, std::vector<int8_t>& vals);
+template int IModelRunner::setInput<int16_t>(std::string input_name, std::vector<int16_t>& vals);
 template int IModelRunner::setInput<int32_t>(std::string input_name, std::vector<int32_t>& vals);
 template int IModelRunner::setInput<int64_t>(std::string input_name, std::vector<int64_t>& vals);
 template int IModelRunner::setInput<unsigned char>(std::string input_name, std::vector<unsigned char>& vals);
 
 template std::vector<float> IModelRunner::getOutput<float>(std::string output_name);
 template std::vector<half_float::half> IModelRunner::getOutput<half_float::half>(std::string output_name);
+template std::vector<int8_t> IModelRunner::getOutput<int8_t>(std::string output_name);
+template std::vector<int16_t> IModelRunner::getOutput<int16_t>(std::string output_name);
 template std::vector<int32_t> IModelRunner::getOutput<int32_t>(std::string output_name);
 template std::vector<int64_t> IModelRunner::getOutput<int64_t>(std::string output_name);
 template std::vector<unsigned char> IModelRunner::getOutput<unsigned char>(std::string output_name);
-- 
cgit v1.2.1