diff options
Diffstat (limited to 'src/backends/reference/workloads/RefDebugWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.hpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp index c1a3e26ec2..2985699f7b 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.hpp +++ b/src/backends/reference/workloads/RefDebugWorkload.hpp @@ -16,6 +16,10 @@ template <armnn::DataType DataType> class RefDebugWorkload : public TypedWorkload<DebugQueueDescriptor, DataType> { public: + RefDebugWorkload(const DebugQueueDescriptor& descriptor, const WorkloadInfo& info) + : TypedWorkload<DebugQueueDescriptor, DataType>(descriptor, info) + , m_Callback(nullptr) {} + static const std::string& GetName() { static const std::string name = std::string("RefDebug") + GetDataTypeName(DataType) + "Workload"; @@ -26,6 +30,11 @@ public: using TypedWorkload<DebugQueueDescriptor, DataType>::TypedWorkload; void Execute() const override; + + void RegisterDebugCallback(const DebugCallbackFunction& func) override; + +private: + DebugCallbackFunction m_Callback; }; using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>; |