2 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp
index a3060f0798..ad52434ef8 100644
--- a/src/backends/reference/RefBackend.cpp
+++ b/src/backends/reference/RefBackend.cpp
@@ -13,6 +13,7 @@
 #include <armnn/backends/IBackendContext.hpp>
 #include <armnn/backends/IMemoryManager.hpp>
 #include <armnn/utility/PolymorphicDowncast.hpp>
+#include <backendsCommon/DefaultAllocator.hpp>
 
 #include <Optimizer.hpp>
 
@@ -86,4 +87,9 @@ void RefBackend::RegisterTensorHandleFactories(class TensorHandleFactoryRegistry
     registry.RegisterFactory(std::make_unique<RefTensorHandleFactory>(memoryManager));
 }
 
+std::unique_ptr<ICustomAllocator> RefBackend::GetDefaultAllocator() const
+{
+    return std::make_unique<DefaultAllocator>();
+}
+
 } // namespace armnn
diff --git a/src/backends/reference/RefBackend.hpp b/src/backends/reference/RefBackend.hpp
index 4d4aba980b..6114ce6218 100644
--- a/src/backends/reference/RefBackend.hpp
+++ b/src/backends/reference/RefBackend.hpp
@@ -60,6 +60,8 @@ public:
     {
         return cpuRefCapabilities;
     };
+
+    std::unique_ptr<ICustomAllocator> GetDefaultAllocator() const override;
 };
 
 } // namespace armnn