From 2e4d889fb036d1c0a34503400a3f45cfc6f9f3e1 Mon Sep 17 00:00:00 2001
From: Matthew Sloyan <matthew.sloyan@arm.com>
Date: Tue, 18 Oct 2022 18:02:48 +0100
Subject: Add FP16 support to IModelRunner

 * Added specific FP16 readfromVector and writeToVector methods.
 * Added FP16 support to float readfromVector and writeToVector methods.
 * Added missing reference to IModelRunner::setInput.

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I6b66468737e672afc925ccad4fb710fbb9427c14
---
 reference_model/src/model_runner.cc | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(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 2395a85..5c086e6 100644
--- a/reference_model/src/model_runner.cc
+++ b/reference_model/src/model_runner.cc
@@ -53,7 +53,7 @@ GraphStatus IModelRunner::run()
 }
 
 template <typename T>
-int IModelRunner::setInput(std::string input_name, std::vector<T> vals)
+int IModelRunner::setInput(std::string input_name, std::vector<T>& vals)
 {
     return model_runner_impl->setInput<T>(input_name, vals);
 }
@@ -65,12 +65,14 @@ std::vector<T> IModelRunner::getOutput(std::string output_name)
 }
 
 // Template explicit specialization
-template int IModelRunner::setInput<float>(std::string input_name, std::vector<float> 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 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<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<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);
\ No newline at end of file
-- 
cgit v1.2.1