commit 549cb7ad5fe0686a5216557ef7ee21b520bf7936
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri Jul 10 17:50:53 2020 +0100
committer Jim Flynn <jim.flynn@arm.com> Mon Jul 13 09:17:43 2020 +0000
tree 34c1ff0c8933cb72129d8b8d30d6e814c272841c
parent 87d0bda9b49d9df4455f1887027e5ead2527c27e

IVGCVSW-5078 Fix segmentation fault on Yolo V3 Big App

 * Register TensorHandleFactories when create Workload Factory

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I6217c6f00d27b67e42b8043cbaf72d9db5789bfb

src/backends/reference/RefBackend.cpp

diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp
index 590fde3..e93b317 100644
--- a/src/backends/reference/RefBackend.cpp
+++ b/src/backends/reference/RefBackend.cpp
@@ -37,6 +37,7 @@
     auto memoryManager = std::make_shared<RefMemoryManager>();
 
     tensorHandleFactoryRegistry.RegisterMemoryManager(memoryManager);
+    tensorHandleFactoryRegistry.RegisterFactory(std::make_unique<RefTensorHandleFactory>(memoryManager));
 
     return std::make_unique<RefWorkloadFactory>(PolymorphicPointerDowncast<RefMemoryManager>(memoryManager));
 }