diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-28 10:29:12 +0000 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-29 15:43:18 +0000 |
commit | 317cae526ba7b58891b6f054bb10af3b6e22d5a3 (patch) | |
tree | e2939198b33e2db223d83f2d9f8ad6c1704e385b /src/backends/reference/workloads/RefDebugWorkload.hpp | |
parent | 83add2165b680b3cf38403a7ce90ea86febd4cc7 (diff) | |
download | armnn-317cae526ba7b58891b6f054bb10af3b6e22d5a3.tar.gz |
IVGCVSW-2866 Implement RegisterDebugCallback for RefDebugWorkload
Change-Id: I9144fb6b7d05561b5b8fd9db5dbe31c9257f10ca
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
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>; |