Add support for boolean outputs in model runner

Comparison operators produce boolean outputs, which need to be written into client data

Allow subgraph traverser to use main block to look for tensors when serialization handler is missing

Signed-off-by: Jiacheng Liang <jiacheng.liang@arm.com>
Change-Id: I6f9af470185541fa6466b3f7786c48f1555fa6f6

1 files changed, 6 insertions, 0 deletions

diff --git a/reference_model/src/model_runner_impl.cc b/reference_model/src/model_runner_impl.cc
index 8089a1a..31e100a 100644
--- a/reference_model/src/model_runner_impl.cc
+++ b/reference_model/src/model_runner_impl.cc
@@ -264,6 +264,12 @@ int ModelRunnerImpl::getOutput(std::string output_name, uint8_t* raw_ptr, size_t
             status             = tensor->writeToVector(ArrayProxy(elements, typed_ptr));
             break;
         }
+        case TOSA_REF_TYPE_BOOL: {
+            auto typed_ptr     = reinterpret_cast<unsigned char*>(raw_ptr);
+            const int elements = size / sizeof(unsigned char);
+            status             = tensor->writeToVector(ArrayProxy(elements, typed_ptr));
+            break;
+        }
         default:
             status = 1;
     }