diff options
Diffstat (limited to 'src/backends/cl/ClBackend.cpp')
| -rw-r--r-- | src/backends/cl/ClBackend.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/cl/ClBackend.cpp b/src/backends/cl/ClBackend.cpp index dd58e002be..5c582695e5 100644 --- a/src/backends/cl/ClBackend.cpp +++ b/src/backends/cl/ClBackend.cpp @@ -5,6 +5,7 @@ #include "ClBackend.hpp" #include "ClBackendContext.hpp" +#include "ClBackendDefaultAllocator.hpp" #include "ClBackendId.hpp" #include "ClBackendModelContext.hpp" #include "ClImportTensorHandleFactory.hpp" @@ -216,6 +217,11 @@ IBackendInternal::ILayerSupportSharedPtr ClBackend::GetLayerSupport(const ModelO return layerSupport; } +std::unique_ptr<ICustomAllocator> ClBackend::GetDefaultAllocator() const +{ + return std::make_unique<ClBackendDefaultAllocator>(); +} + OptimizationViews ClBackend::OptimizeSubgraphView(const SubgraphView& subgraph, const ModelOptions& modelOptions) const { |