From 5c54c3874ff5dc1656b9f28288e46086586f21b9 Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Wed, 9 Nov 2022 16:28:51 +0000 Subject: IVGCVSW-7165 Implement TosaRefPreCompiledWorkload::Execute() * Added FP32 support for TOSA Reference Backend. * Added main block creation to OptimizeSubgraphView, this will only occur once. Change-Id: I169dac50b78e2c693da6327962c9f1d3ae3bd712 Signed-off-by: James Conroy Signed-off-by: Matthew Sloyan --- src/backends/tosaReference/TosaRefBackend.cpp | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'src/backends/tosaReference/TosaRefBackend.cpp') diff --git a/src/backends/tosaReference/TosaRefBackend.cpp b/src/backends/tosaReference/TosaRefBackend.cpp index 688cf93b49..e3a516a5f9 100644 --- a/src/backends/tosaReference/TosaRefBackend.cpp +++ b/src/backends/tosaReference/TosaRefBackend.cpp @@ -85,6 +85,9 @@ OptimizationViews TosaRefBackend::OptimizeSubgraphView(const SubgraphView& subgr OptimizationViews optimizationViews(modelOptions); auto handler = std::make_unique(); + // A main block should only be added once. + bool isMain = true; + auto it = subgraph.endIConnectable(); while (it != subgraph.beginIConnectable()) { @@ -97,8 +100,13 @@ OptimizationViews TosaRefBackend::OptimizeSubgraphView(const SubgraphView& subgr continue; } - tosa::TosaSerializationBasicBlock* mappings = GetTosaMappingFromLayer(&base); + tosa::TosaSerializationBasicBlock* mappings = GetTosaMappingFromLayer(&base, isMain); handler.get()->GetBlocks().push_back(mappings); + + if(isMain) + { + isMain = false; + } } auto compiledBlob = -- cgit v1.2.1