diff options
| author | Keith Davis <keith.davis@arm.com> | 2021-08-18 17:14:05 +0100 |
|---|---|---|
| committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2021-08-20 09:10:22 +0000 |
| commit | 4914d0c1f4cbb01fe276a7093af4cff13270b74a (patch) | |
| tree | ce6c4364bb596803148833af535215aa6ab6de90 /include | |
| parent | e369dbddae66c58f4b444c4b25871c10af19ed9d (diff) | |
| download | armnn-4914d0c1f4cbb01fe276a7093af4cff13270b74a.tar.gz | |
IVGCVSW-6249 Add ProfilingDetails Macros to all workloads in Ref, Neon, CL
* Add functionality to only output network details in ExNet
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I0c45e67193f308ce7b86f1bb1a918a266fefba2e
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/IProfiler.hpp | 4 | ||||
| -rw-r--r-- | include/armnn/IRuntime.hpp | 10 | ||||
| -rw-r--r-- | include/armnn/Types.hpp | 9 |
3 files changed, 16 insertions, 7 deletions
diff --git a/include/armnn/IProfiler.hpp b/include/armnn/IProfiler.hpp index 5fbb67000a..1b450d0151 100644 --- a/include/armnn/IProfiler.hpp +++ b/include/armnn/IProfiler.hpp @@ -40,8 +40,8 @@ public: void Print(std::ostream& outStream) const; /// Print out details of each layer within the network that possesses a descriptor. - /// Also outputs tensor info. - void EnableNetworkDetailsToStdOut(); + /// Also outputs tensor info. This will be part of the profiling json output + void EnableNetworkDetailsToStdOut(ProfilingDetailsMethod detailsMethod); ~IProfiler(); IProfiler(); diff --git a/include/armnn/IRuntime.hpp b/include/armnn/IRuntime.hpp index 97a9c2889e..345cdeb10a 100644 --- a/include/armnn/IRuntime.hpp +++ b/include/armnn/IRuntime.hpp @@ -40,7 +40,7 @@ struct INetworkProperties m_ExportEnabled(exportEnabled), m_AsyncEnabled(asyncEnabled), m_ProfilingEnabled(profilingEnabled), - m_OutputNetworkDetails(false), + m_OutputNetworkDetailsMethod(ProfilingDetailsMethod::Undefined), m_InputSource(m_ImportEnabled ? MemorySource::Malloc : MemorySource::Undefined), m_OutputSource(m_ExportEnabled ? MemorySource::Malloc : MemorySource::Undefined) {} @@ -55,7 +55,7 @@ struct INetworkProperties m_ExportEnabled(outputSource != MemorySource::Undefined), m_AsyncEnabled(asyncEnabled), m_ProfilingEnabled(profilingEnabled), - m_OutputNetworkDetails(false), + m_OutputNetworkDetailsMethod(ProfilingDetailsMethod::Undefined), m_InputSource(inputSource), m_OutputSource(outputSource) { @@ -66,12 +66,12 @@ struct INetworkProperties MemorySource inputSource, MemorySource outputSource, bool profilingEnabled = false, - bool outputDetails = false) + ProfilingDetailsMethod detailsMethod = ProfilingDetailsMethod::Undefined) : m_ImportEnabled(inputSource != MemorySource::Undefined), m_ExportEnabled(outputSource != MemorySource::Undefined), m_AsyncEnabled(asyncEnabled), m_ProfilingEnabled(profilingEnabled), - m_OutputNetworkDetails(outputDetails), + m_OutputNetworkDetailsMethod(detailsMethod), m_InputSource(inputSource), m_OutputSource(outputSource) {} @@ -85,7 +85,7 @@ struct INetworkProperties const bool m_ProfilingEnabled; - const bool m_OutputNetworkDetails; + const ProfilingDetailsMethod m_OutputNetworkDetailsMethod; const MemorySource m_InputSource; const MemorySource m_OutputSource; diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 056aa83d2f..5e00026eba 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -56,6 +56,15 @@ enum class DataLayout NHWC = 2 }; +/// Define the behaviour of the internal profiler when outputting network details +enum class ProfilingDetailsMethod +{ + Undefined = 0, + DetailsWithEvents = 1, + DetailsOnly = 2 +}; + + enum class QosExecPriority { Low = 0, |