diff options
Diffstat (limited to 'delegate/include/DelegateOptions.hpp')
| -rw-r--r-- | delegate/include/DelegateOptions.hpp | 58 |
1 files changed, 57 insertions, 1 deletions
diff --git a/delegate/include/DelegateOptions.hpp b/delegate/include/DelegateOptions.hpp index ace0859f7e..b8847440ed 100644 --- a/delegate/include/DelegateOptions.hpp +++ b/delegate/include/DelegateOptions.hpp @@ -41,6 +41,12 @@ public: void SetBackends(const std::vector<armnn::BackendId>& backends) { m_Backends = backends; } + void SetDynamicBackendsPath(const std::string& dynamicBackendsPath) { m_DynamicBackendsPath = dynamicBackendsPath; } + const std::string& GetDynamicBackendsPath() const { return m_DynamicBackendsPath; } + + void SetGpuProfilingState(bool gpuProfilingState) { m_EnableGpuProfiling = gpuProfilingState; } + bool GetGpuProfilingState() { return m_EnableGpuProfiling; } + const std::vector<armnn::BackendOptions>& GetBackendOptions() const { return m_BackendOptions; } /// Appends a backend option to the list of backend options @@ -62,6 +68,23 @@ public: const armnn::Optional<armnn::DebugCallbackFunction>& GetDebugCallbackFunction() const { return m_DebugCallbackFunc; } + void SetInternalProfilingParams(bool internalProfilingState, + const armnn::ProfilingDetailsMethod& internalProfilingDetail) + { m_InternalProfilingEnabled = internalProfilingState; m_InternalProfilingDetail = internalProfilingDetail; } + + bool GetInternalProfilingState() const { return m_InternalProfilingEnabled; } + const armnn::ProfilingDetailsMethod& GetInternalProfilingDetail() const { return m_InternalProfilingDetail; } + + void SetExternalProfilingParams( + const armnn::IRuntime::CreationOptions::ExternalProfilingOptions& externalProfilingParams) + { m_ProfilingOptions = externalProfilingParams; } + + const armnn::IRuntime::CreationOptions::ExternalProfilingOptions& GetExternalProfilingParams() const + { return m_ProfilingOptions; } + + void SetSerializeToDot(const std::string& serializeToDotFile) { m_SerializeToDot = serializeToDotFile; } + const std::string& GetSerializeToDot() const { return m_SerializeToDot; } + private: /// Which backend to run Delegate on. /// Examples of possible values are: CpuRef, CpuAcc, GpuAcc. @@ -87,6 +110,13 @@ private: /// "KernelProfilingEnabled" : bool [true | false] std::vector<armnn::BackendOptions> m_BackendOptions; + /// Dynamic backend path. + /// This is the directory that will be searched for any dynamic backends. + std::string m_DynamicBackendsPath = ""; + + /// Enable Gpu Profiling. + bool m_EnableGpuProfiling = false; + /// OptimizerOptions /// Reduce Fp32 data to Fp16 for faster processing /// bool m_ReduceFp32ToFp16; @@ -100,13 +130,39 @@ private: /// ModelOptions m_ModelOptions; armnn::OptimizerOptions m_OptimizerOptions; + /// External profiling options. + /// Indicates whether external profiling is enabled or not. + /// bool m_EnableProfiling + /// Indicates whether external timeline profiling is enabled or not. + /// bool m_TimelineEnabled + /// Path to a file in which outgoing timeline profiling messages will be stored. + /// std::string m_OutgoingCaptureFile + /// Path to a file in which incoming timeline profiling messages will be stored. + /// std::string m_IncomingCaptureFile + /// Enable profiling output to file only. + /// bool m_FileOnly + /// The duration at which captured profiling messages will be flushed. + /// uint32_t m_CapturePeriod + /// The format of the file used for outputting profiling data. + /// std::string m_FileFormat + armnn::IRuntime::CreationOptions::ExternalProfilingOptions m_ProfilingOptions; + + /// Internal profiling options. + /// Indicates whether internal profiling is enabled or not. + bool m_InternalProfilingEnabled = false; + /// Sets the level of detail output by the profiling. Options are DetailsWithEvents = 1 and DetailsOnly = 2 + armnn::ProfilingDetailsMethod m_InternalProfilingDetail = armnn::ProfilingDetailsMethod::DetailsWithEvents; + /// Severity level for logging within ArmNN that will be used on creation of the delegate armnn::Optional<armnn::LogSeverity> m_LoggingSeverity; /// A callback function to debug layers performing custom computations on intermediate tensors. /// If a function is not registered, and debug is enabled in OptimizerOptions, /// debug will print information of the intermediate tensors. - armnn::Optional<armnn::DebugCallbackFunction> m_DebugCallbackFunc; + armnn::Optional<armnn::DebugCallbackFunction> m_DebugCallbackFunc; + + /// If not empty then the optimized model will be serialized to a file with this file name in "dot" format. + std::string m_SerializeToDot = ""; }; } // namespace armnnDelegate |