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.cpp | |
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.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.cpp | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefDebugWorkload.cpp b/src/backends/reference/workloads/RefDebugWorkload.cpp index 412d399adc..3da925913a 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.cpp +++ b/src/backends/reference/workloads/RefDebugWorkload.cpp @@ -9,6 +9,8 @@ #include <TypeUtils.hpp> +#include <cstring> + namespace armnn { @@ -20,12 +22,27 @@ void RefDebugWorkload<DataType>::Execute() const ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, GetName() + "_Execute"); const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); - const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]); const T* inputData = GetInputTensorData<T>(0, m_Data); T* outputData = GetOutputTensorData<T>(0, m_Data); - Debug(inputInfo, outputInfo, inputData, outputData, m_Data.m_Guid, m_Data.m_LayerName, m_Data.m_SlotIndex); + if (m_Callback) + { + m_Callback(m_Data.m_Guid, m_Data.m_SlotIndex, m_Data.m_Inputs[0]); + } + else + { + Debug(inputInfo, inputData, m_Data.m_Guid, m_Data.m_LayerName, m_Data.m_SlotIndex); + } + + std::memcpy(outputData, inputData, inputInfo.GetNumElements()*sizeof(T)); + +} + +template<armnn::DataType DataType> +void RefDebugWorkload<DataType>::RegisterDebugCallback(const DebugCallbackFunction& func) +{ + m_Callback = func; } template class RefDebugWorkload<DataType::Float32>; |