diff options
Diffstat (limited to 'tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp')
| -rw-r--r-- | tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | 279 |
1 files changed, 226 insertions, 53 deletions
diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp index bc2868ab35..14d7fe5551 100644 --- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp +++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp @@ -1,63 +1,83 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once -#include <armnn/IRuntime.hpp> -#include <armnn/Types.hpp> #include <armnn/Logging.hpp> #include <armnn/utility/StringUtils.hpp> -#include <armnnUtils/TContainer.hpp> +#include <armnn/utility/NumericCast.hpp> +#include <armnn/BackendRegistry.hpp> #include <iostream> #include <fstream> +#include <iomanip> +#include <iterator> +/** + * Given a measured duration and a threshold time tell the user whether we succeeded or not. + * + * @param duration the measured inference duration. + * @param thresholdTime the threshold time in milliseconds. + * @return false if the measured time exceeded the threshold. + */ +bool CheckInferenceTimeThreshold(const std::chrono::duration<double, std::milli>& duration, + const double& thresholdTime); -std::vector<unsigned int> ParseArray(std::istream& stream); - -/// Splits a given string at every accurance of delimiter into a vector of string -std::vector<std::string> ParseStringList(const std::string& inputString, const char* delimiter); - -struct TensorPrinter +inline bool CheckRequestedBackendsAreValid(const std::vector<armnn::BackendId>& backendIds, + armnn::Optional<std::string&> invalidBackendIds = armnn::EmptyOptional()) { - TensorPrinter(const std::string& binding, - const armnn::TensorInfo& info, - const std::string& outputTensorFile, - bool dequantizeOutput, - bool printToConsole = true); - - void operator()(const std::vector<float>& values); - - void operator()(const std::vector<uint8_t>& values); - - void operator()(const std::vector<int>& values); + if (backendIds.empty()) + { + return false; + } - void operator()(const std::vector<int8_t>& values); + armnn::BackendIdSet validBackendIds = armnn::BackendRegistryInstance().GetBackendIds(); -private: - template<typename Container, typename Delegate> - void ForEachValue(const Container& c, Delegate delegate); + bool allValid = true; + for (const auto& backendId : backendIds) + { + if (std::find(validBackendIds.begin(), validBackendIds.end(), backendId) == validBackendIds.end()) + { + allValid = false; + if (invalidBackendIds) + { + if (!invalidBackendIds.value().empty()) + { + invalidBackendIds.value() += ", "; + } + invalidBackendIds.value() += backendId; + } + } + } + return allValid; +} - template<typename T> - void WriteToFile(const std::vector<T>& values); +std::vector<unsigned int> ParseArray(std::istream& stream); - std::string m_OutputBinding; - float m_Scale; - int m_Offset; - std::string m_OutputTensorFile; - bool m_DequantizeOutput; - bool m_PrintToConsole; -}; +/// Splits a given string at every accurance of delimiter into a vector of string +std::vector<std::string> ParseStringList(const std::string& inputString, const char* delimiter); -using QuantizationParams = std::pair<float, int32_t>; +/// Dequantize an array of a given type +/// @param array Type erased array to dequantize +/// @param numElements Elements in the array +/// @param array Type erased array to dequantize +template <typename T> +std::vector<float> DequantizeArray(const void* array, unsigned int numElements, float scale, int32_t offset) +{ + const T* quantizedArray = reinterpret_cast<const T*>(array); + std::vector<float> dequantizedVector; + dequantizedVector.reserve(numElements); + for (unsigned int i = 0; i < numElements; ++i) + { + float f = armnn::Dequantize(*(quantizedArray + i), scale, offset); + dequantizedVector.push_back(f); + } + return dequantizedVector; +} -void PopulateTensorWithData(armnnUtils::TContainer& tensorData, - unsigned int numElements, - const std::string& dataTypeStr, - const armnn::Optional<QuantizationParams>& qParams, - const armnn::Optional<std::string>& dataFile); +void LogAndThrow(std::string eMsg); /** * Verifies if the given string is a valid path. Reports invalid paths to std::err. @@ -75,6 +95,152 @@ bool ValidatePath(const std::string& file, const bool expectFile); * */ bool ValidatePaths(const std::vector<std::string>& fileVec, const bool expectFile); +/// Returns a function of read the given type as a string +template <typename Integer, typename std::enable_if_t<std::is_integral<Integer>::value>* = nullptr> +std::function<Integer(const std::string&)> GetParseElementFunc() +{ + return [](const std::string& s) { return armnn::numeric_cast<Integer>(std::stoi(s)); }; +} + +template <typename Float, std::enable_if_t<std::is_floating_point<Float>::value>* = nullptr> +std::function<Float(const std::string&)> GetParseElementFunc() +{ + return [](const std::string& s) { return std::stof(s); }; +} + +template <typename T> +void PopulateTensorWithData(T* tensor, + const unsigned int numElements, + const armnn::Optional<std::string>& dataFile, + const std::string& inputName) +{ + const bool readFromFile = dataFile.has_value() && !dataFile.value().empty(); + + std::ifstream inputTensorFile; + if (!readFromFile) + { + std::fill(tensor, tensor + numElements, 0); + return; + } + else + { + inputTensorFile = std::ifstream(dataFile.value()); + } + + auto parseElementFunc = GetParseElementFunc<T>(); + std::string line; + unsigned int index = 0; + while (std::getline(inputTensorFile, line)) + { + std::vector<std::string> tokens = armnn::stringUtils::StringTokenizer(line, "\t ,:"); + for (const std::string& token : tokens) + { + if (!token.empty()) // See https://stackoverflow.com/questions/10437406/ + { + try + { + if (index == numElements) + { + ARMNN_LOG(error) << "Number of elements: " << (index +1) << " in file \"" << dataFile.value() + << "\" does not match number of elements: " << numElements + << " for input \"" << inputName << "\"."; + } + *(tensor + index) = parseElementFunc(token); + index++; + } + catch (const std::exception&) + { + ARMNN_LOG(error) << "'" << token << "' is not a valid number. It has been ignored."; + } + } + } + } + + if (index != numElements) + { + ARMNN_LOG(error) << "Number of elements: " << (index +1) << " in file \"" << inputName + << "\" does not match number of elements: " << numElements + << " for input \"" << inputName << "\"."; + } +} + +template<typename T> +void WriteToFile(const std::string& outputTensorFileName, + const std::string& outputName, + const T* const array, + const unsigned int numElements) +{ + std::ofstream outputTensorFile; + outputTensorFile.open(outputTensorFileName, std::ofstream::out | std::ofstream::trunc); + if (outputTensorFile.is_open()) + { + outputTensorFile << outputName << ": "; + std::copy(array, array + numElements, std::ostream_iterator<T>(outputTensorFile, " ")); + } + else + { + ARMNN_LOG(info) << "Output Tensor File: " << outputTensorFileName << " could not be opened!"; + } + outputTensorFile.close(); +} + +struct OutputWriteInfo +{ + const armnn::Optional<std::string>& m_OutputTensorFile; + const std::string& m_OutputName; + const armnn::Tensor& m_Tensor; + const bool m_PrintTensor; +}; + +template <typename T> +void PrintTensor(OutputWriteInfo& info, const char* formatString) +{ + const T* array = reinterpret_cast<const T*>(info.m_Tensor.GetMemoryArea()); + + if (info.m_OutputTensorFile.has_value()) + { + WriteToFile(info.m_OutputTensorFile.value(), + info.m_OutputName, + array, + info.m_Tensor.GetNumElements()); + } + + if (info.m_PrintTensor) + { + for (unsigned int i = 0; i < info.m_Tensor.GetNumElements(); i++) + { + printf(formatString, array[i]); + } + } +} + +template <typename T> +void PrintQuantizedTensor(OutputWriteInfo& info) +{ + std::vector<float> dequantizedValues; + auto tensor = info.m_Tensor; + dequantizedValues = DequantizeArray<T>(tensor.GetMemoryArea(), + tensor.GetNumElements(), + tensor.GetInfo().GetQuantizationScale(), + tensor.GetInfo().GetQuantizationOffset()); + + if (info.m_OutputTensorFile.has_value()) + { + WriteToFile(info.m_OutputTensorFile.value(), + info.m_OutputName, + dequantizedValues.data(), + tensor.GetNumElements()); + } + + if (info.m_PrintTensor) + { + std::for_each(dequantizedValues.begin(), dequantizedValues.end(), [&](float value) + { + printf("%f ", value); + }); + } +} + template<typename T, typename TParseElementFunc> std::vector<T> ParseArrayImpl(std::istream& stream, TParseElementFunc parseElementFunc, const char* chars = "\t ,:") { @@ -103,21 +269,28 @@ std::vector<T> ParseArrayImpl(std::istream& stream, TParseElementFunc parseEleme return result; } -template <typename T, typename TParseElementFunc> -void PopulateTensorWithDataGeneric(std::vector<T>& tensorData, - unsigned int numElements, - const armnn::Optional<std::string>& dataFile, - TParseElementFunc parseFunction) +/// Compute the root-mean-square error (RMSE) +/// @param expected +/// @param actual +/// @param size size of the tensor +/// @return float the RMSE +template<typename T> +float ComputeRMSE(const void* expected, const void* actual, const size_t size) { - const bool readFromFile = dataFile.has_value() && !dataFile.value().empty(); + auto typedExpected = reinterpret_cast<const T*>(expected); + auto typedActual = reinterpret_cast<const T*>(actual); - std::ifstream inputTensorFile; - if (readFromFile) + T errorSum = 0; + + for (unsigned int i = 0; i < size; i++) { - inputTensorFile = std::ifstream(dataFile.value()); + if (std::abs(typedExpected[i] - typedActual[i]) != 0) + { + std::cout << ""; + } + errorSum += std::pow(std::abs(typedExpected[i] - typedActual[i]), 2); } - tensorData = readFromFile ? - ParseArrayImpl<T>(inputTensorFile, parseFunction) : - std::vector<T>(numElements, static_cast<T>(0)); -} + float rmse = std::sqrt(armnn::numeric_cast<float>(errorSum) / armnn::numeric_cast<float>(size / sizeof(T))); + return rmse; +}
\ No newline at end of file |