diff options
Diffstat (limited to 'include/armnn/IRuntime.hpp')
| -rw-r--r-- | include/armnn/IRuntime.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/include/armnn/IRuntime.hpp b/include/armnn/IRuntime.hpp index f88b6b664f..bee61d21a9 100644 --- a/include/armnn/IRuntime.hpp +++ b/include/armnn/IRuntime.hpp @@ -14,6 +14,7 @@ #include "TypesUtils.hpp" #include "profiling/ILocalPacketHandler.hpp" +#include <armnn/backends/ICustomAllocator.hpp> #include <memory> namespace armnn @@ -88,6 +89,7 @@ public: : m_GpuAccTunedParameters(nullptr) , m_EnableGpuProfiling(false) , m_DynamicBackendsPath("") + , m_CustomAllocator(nullptr) {} /// If set, uses the GpuAcc tuned parameters from the given object when executing GPU workloads. @@ -101,6 +103,11 @@ public: /// Only a single path is allowed for the override std::string m_DynamicBackendsPath; + /// A Custom Allocator used for allocation of working memory in the backends. + /// Set this for when you need to allocate Protected Working Memory, required for ProtectedMode + /// Only supported for GpuAcc + ICustomAllocator* m_CustomAllocator; + struct ExternalProfilingOptions { ExternalProfilingOptions() |