From 81db5d2f275f69cc0d3e8687af57bdba99971042 Mon Sep 17 00:00:00 2001
From: Eric Kunze <eric.kunze@arm.com>
Date: Mon, 11 Dec 2023 16:51:25 -0800
Subject: Add interface to model runner to run a basic block

Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: Ife703534594713616a992c460d201a8666c7b770
---
 reference_model/include/model_runner.h | 1 +
 reference_model/src/model_runner.cc    | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

(limited to 'reference_model')

diff --git a/reference_model/include/model_runner.h b/reference_model/include/model_runner.h
index 2c610b8..83e4931 100644
--- a/reference_model/include/model_runner.h
+++ b/reference_model/include/model_runner.h
@@ -52,6 +52,7 @@ public:
      * The status of the graph will be returned upon completion.
      */
     GraphStatus initialize(tosa::TosaSerializationHandler& serialization_handler);
+    GraphStatus initialize(tosa::TosaSerializationBasicBlock& basic_block);
 
     /*
      * Run the model using the internal SubgraphTraverser created during initialization.
diff --git a/reference_model/src/model_runner.cc b/reference_model/src/model_runner.cc
index 0ef44f9..6f65202 100644
--- a/reference_model/src/model_runner.cc
+++ b/reference_model/src/model_runner.cc
@@ -47,6 +47,11 @@ GraphStatus IModelRunner::initialize(tosa::TosaSerializationHandler& serializati
     return model_runner_impl->initialize(serialization_handler);
 }
 
+GraphStatus IModelRunner::initialize(tosa::TosaSerializationBasicBlock& basic_block)
+{
+    return model_runner_impl->initialize(basic_block);
+}
+
 GraphStatus IModelRunner::run()
 {
     return model_runner_impl->run();
@@ -85,4 +90,4 @@ template std::vector<float> IModelRunner::getOutput<float>(std::string output_na
 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
+template std::vector<unsigned char> IModelRunner::getOutput<unsigned char>(std::string output_name);
-- 
cgit v1.2.1