diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/cl/ClBackend.cpp | 5 | ||||
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 5 | ||||
-rw-r--r-- | src/backends/reference/RefBackend.cpp | 5 |
3 files changed, 12 insertions, 3 deletions
diff --git a/src/backends/cl/ClBackend.cpp b/src/backends/cl/ClBackend.cpp index 840da8bda3..95acf009de 100644 --- a/src/backends/cl/ClBackend.cpp +++ b/src/backends/cl/ClBackend.cpp @@ -4,7 +4,10 @@ // #include "ClBackend.hpp" +#include "ClWorkloadFactory.hpp" + #include <backends/BackendRegistry.hpp> + #include <boost/cast.hpp> namespace armnn @@ -36,7 +39,7 @@ const ILayerSupport& ClBackend::GetLayerSupport() const std::unique_ptr<IWorkloadFactory> ClBackend::CreateWorkloadFactory() const { - return nullptr; + return std::make_unique<ClWorkloadFactory>(); } void ClBackend::Destroy(IBackend* backend) diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index e35bf7ea45..2e235b6c2a 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -4,7 +4,10 @@ // #include "NeonBackend.hpp" +#include "NeonWorkloadFactory.hpp" + #include <backends/BackendRegistry.hpp> + #include <boost/cast.hpp> namespace armnn @@ -37,7 +40,7 @@ const ILayerSupport& NeonBackend::GetLayerSupport() const std::unique_ptr<IWorkloadFactory> NeonBackend::CreateWorkloadFactory() const { - return nullptr; + return std::make_unique<NeonWorkloadFactory>(); } void NeonBackend::Destroy(IBackend* backend) diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp index 63eff52b75..ef52a5edeb 100644 --- a/src/backends/reference/RefBackend.cpp +++ b/src/backends/reference/RefBackend.cpp @@ -4,7 +4,10 @@ // #include "RefBackend.hpp" +#include "RefWorkloadFactory.hpp" + #include <backends/BackendRegistry.hpp> + #include <boost/cast.hpp> namespace armnn @@ -36,7 +39,7 @@ const ILayerSupport& RefBackend::GetLayerSupport() const std::unique_ptr<IWorkloadFactory> RefBackend::CreateWorkloadFactory() const { - return nullptr; + return std::make_unique<RefWorkloadFactory>(); } void RefBackend::Destroy(IBackend* backend) |