diff options
Diffstat (limited to 'src/backends/cl/test/ClWorkloadFactoryHelper.hpp')
-rw-r--r-- | src/backends/cl/test/ClWorkloadFactoryHelper.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/cl/test/ClWorkloadFactoryHelper.hpp b/src/backends/cl/test/ClWorkloadFactoryHelper.hpp index 6e3c6fc05a..f7f1629b27 100644 --- a/src/backends/cl/test/ClWorkloadFactoryHelper.hpp +++ b/src/backends/cl/test/ClWorkloadFactoryHelper.hpp @@ -27,9 +27,12 @@ struct WorkloadFactoryHelper<armnn::ClWorkloadFactory> } static armnn::ClWorkloadFactory GetFactory( - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ModelOptions& modelOptions = {}) { - return armnn::ClWorkloadFactory(armnn::PolymorphicPointerDowncast<armnn::ClMemoryManager>(memoryManager)); + armnn::ClBackend backend; + return armnn::ClWorkloadFactory(armnn::PolymorphicPointerDowncast<armnn::ClMemoryManager>(memoryManager), + backend.CreateBackendSpecificModelContext(modelOptions)); } static armnn::ClTensorHandleFactory GetTensorHandleFactory( |