diff options
author | Eric Kunze <eric.kunze@arm.com> | 2023-12-11 16:51:25 -0800 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-12-12 22:27:35 +0000 |
commit | 81db5d2f275f69cc0d3e8687af57bdba99971042 (patch) | |
tree | c81e5e055c075446295b2c5b169df074653ac63b | |
parent | 598d74af92914ab3fb02b7f9c3a84aea09226818 (diff) | |
download | reference_model-81db5d2f275f69cc0d3e8687af57bdba99971042.tar.gz |
Add interface to model runner to run a basic block
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: Ife703534594713616a992c460d201a8666c7b770
-rw-r--r-- | reference_model/include/model_runner.h | 1 | ||||
-rw-r--r-- | reference_model/src/model_runner.cc | 7 |
2 files changed, 7 insertions, 1 deletions
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); |