aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTestUtils
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnTestUtils')
-rwxr-xr-xsrc/armnnTestUtils/CMakeLists.txt5
-rw-r--r--src/armnnTestUtils/MockBackend.cpp6
-rw-r--r--src/armnnTestUtils/MockWorkloadFactoryHelper.hpp41
-rw-r--r--src/armnnTestUtils/TensorHelpers.hpp235
-rw-r--r--src/armnnTestUtils/UnitTests.hpp2
5 files changed, 9 insertions, 280 deletions
diff --git a/src/armnnTestUtils/CMakeLists.txt b/src/armnnTestUtils/CMakeLists.txt
index cab0ba353e..e33fed77d7 100755
--- a/src/armnnTestUtils/CMakeLists.txt
+++ b/src/armnnTestUtils/CMakeLists.txt
@@ -11,10 +11,11 @@ list(APPEND armnnTestUtils_sources
../../include/armnnTestUtils/MockBackend.hpp
../../include/armnnTestUtils/MockMemoryManager.hpp
../../include/armnnTestUtils/MockTensorHandle.hpp
+ ../../include/armnnTestUtils/MockWorkloadFactoryHelper.hpp
../../include/armnnTestUtils/PredicateResult.hpp
../../include/armnnTestUtils/TensorCopyUtils.hpp
../../include/armnnTestUtils/WorkloadTestUtils.hpp
- TensorHelpers.hpp
+ ../../include/armnnTestUtils/TensorHelpers.hpp
CreateWorkload.hpp
CommonTestUtils.cpp
CommonTestUtils.hpp
@@ -24,9 +25,7 @@ list(APPEND armnnTestUtils_sources
MockBackend.cpp
MockMemoryManager.cpp
MockTensorHandle.cpp
- MockTensorHandleFactory.hpp
MockTensorHandleFactory.cpp
- MockWorkloadFactoryHelper.hpp
TensorCopyUtils.cpp
TestUtils.cpp
TestUtils.hpp
diff --git a/src/armnnTestUtils/MockBackend.cpp b/src/armnnTestUtils/MockBackend.cpp
index 40f97cd28e..29996bfd26 100644
--- a/src/armnnTestUtils/MockBackend.cpp
+++ b/src/armnnTestUtils/MockBackend.cpp
@@ -15,6 +15,12 @@ constexpr const char* MockBackendId()
return "CpuMock";
}
+const BackendId& MockBackend::GetIdStatic()
+{
+ static const BackendId s_Id{MockBackendId()};
+ return s_Id;
+}
+
namespace
{
static const BackendId s_Id{ MockBackendId() };
diff --git a/src/armnnTestUtils/MockWorkloadFactoryHelper.hpp b/src/armnnTestUtils/MockWorkloadFactoryHelper.hpp
deleted file mode 100644
index 6b61cf6474..0000000000
--- a/src/armnnTestUtils/MockWorkloadFactoryHelper.hpp
+++ /dev/null
@@ -1,41 +0,0 @@
-//
-// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-
-#pragma once
-
-#include <MockTensorHandleFactory.hpp>
-#include <armnnTestUtils/MockBackend.hpp>
-#include <backendsCommon/test/WorkloadFactoryHelper.hpp>
-
-namespace
-{
-
-template <>
-struct WorkloadFactoryHelper<armnn::MockWorkloadFactory>
-{
- static armnn::IBackendInternal::IMemoryManagerSharedPtr GetMemoryManager()
- {
- armnn::MockBackend backend;
- return backend.CreateMemoryManager();
- }
-
- static armnn::MockWorkloadFactory
- GetFactory(const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr)
- {
- IgnoreUnused(memoryManager);
- return armnn::MockWorkloadFactory();
- }
-
- static armnn::MockTensorHandleFactory
- GetTensorHandleFactory(const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr)
- {
-
- return armnn::MockTensorHandleFactory(std::static_pointer_cast<armnn::MockMemoryManager>(memoryManager));
- }
-};
-
-using MockWorkloadFactoryHelper = WorkloadFactoryHelper<armnn::MockWorkloadFactory>;
-
-} // anonymous namespace
diff --git a/src/armnnTestUtils/TensorHelpers.hpp b/src/armnnTestUtils/TensorHelpers.hpp
deleted file mode 100644
index d51e4b1bce..0000000000
--- a/src/armnnTestUtils/TensorHelpers.hpp
+++ /dev/null
@@ -1,235 +0,0 @@
-//
-// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-#pragma once
-
-#include <armnnTestUtils/PredicateResult.hpp>
-
-#include <armnn/Tensor.hpp>
-#include <armnn/utility/Assert.hpp>
-#include <armnnUtils/FloatingPointComparison.hpp>
-
-#include <QuantizeHelper.hpp>
-
-#include <doctest/doctest.h>
-
-#include <array>
-#include <cmath>
-#include <random>
-#include <vector>
-
-constexpr float g_FloatCloseToZeroTolerance = 1.0e-6f;
-
-template<typename T, bool isQuantized = true>
-struct SelectiveComparer
-{
- static bool Compare(T a, T b)
- {
- return (std::max(a, b) - std::min(a, b)) <= 1;
- }
-
-};
-
-template<typename T>
-struct SelectiveComparer<T, false>
-{
- static bool Compare(T a, T b)
- {
- // If a or b is zero, percent_tolerance does an exact match, so compare to a small, constant tolerance instead.
- if (a == 0.0f || b == 0.0f)
- {
- return std::abs(a - b) <= g_FloatCloseToZeroTolerance;
- }
-
- if (std::isinf(a) && a == b)
- {
- return true;
- }
-
- if (std::isnan(a) && std::isnan(b))
- {
- return true;
- }
-
- // For unquantized floats we use a tolerance of 1%.
- return armnnUtils::within_percentage_tolerance(a, b);
- }
-};
-
-template<typename T>
-bool SelectiveCompare(T a, T b)
-{
- return SelectiveComparer<T, armnn::IsQuantizedType<T>()>::Compare(a, b);
-};
-
-template<typename T>
-bool SelectiveCompareBoolean(T a, T b)
-{
- return (((a == 0) && (b == 0)) || ((a != 0) && (b != 0)));
-};
-
-template <typename T>
-armnn::PredicateResult CompareTensors(const std::vector<T>& actualData,
- const std::vector<T>& expectedData,
- const armnn::TensorShape& actualShape,
- const armnn::TensorShape& expectedShape,
- bool compareBoolean = false,
- bool isDynamic = false)
-{
- if (actualData.size() != expectedData.size())
- {
- armnn::PredicateResult res(false);
- res.Message() << "Different data size ["
- << actualData.size()
- << "!="
- << expectedData.size()
- << "]";
- return res;
- }
-
- if (actualShape.GetNumDimensions() != expectedShape.GetNumDimensions())
- {
- armnn::PredicateResult res(false);
- res.Message() << "Different number of dimensions ["
- << actualShape.GetNumDimensions()
- << "!="
- << expectedShape.GetNumDimensions()
- << "]";
- return res;
- }
-
- if (actualShape.GetNumElements() != expectedShape.GetNumElements())
- {
- armnn::PredicateResult res(false);
- res.Message() << "Different number of elements ["
- << actualShape.GetNumElements()
- << "!="
- << expectedShape.GetNumElements()
- << "]";
- return res;
- }
-
- unsigned int numberOfDimensions = actualShape.GetNumDimensions();
-
- if (!isDynamic)
- {
- // Checks they are same shape.
- for (unsigned int i = 0; i < numberOfDimensions; ++i)
- {
- if (actualShape[i] != expectedShape[i])
- {
- armnn::PredicateResult res(false);
- res.Message() << "Different shapes ["
- << actualShape[i]
- << "!="
- << expectedShape[i]
- << "]";
- return res;
- }
- }
- }
-
- // Fun iteration over n dimensions.
- std::vector<unsigned int> indices;
- for (unsigned int i = 0; i < numberOfDimensions; i++)
- {
- indices.emplace_back(0);
- }
-
- std::stringstream errorString;
- int numFailedElements = 0;
- constexpr int maxReportedDifferences = 3;
- unsigned int index = 0;
-
- // Compare data element by element.
- while (true)
- {
- bool comparison;
- // As true for uint8_t is non-zero (1-255) we must have a dedicated compare for Booleans.
- if(compareBoolean)
- {
- comparison = SelectiveCompareBoolean(actualData[index], expectedData[index]);
- }
- else
- {
- comparison = SelectiveCompare(actualData[index], expectedData[index]);
- }
-
- if (!comparison)
- {
- ++numFailedElements;
-
- if (numFailedElements <= maxReportedDifferences)
- {
- if (numFailedElements >= 2)
- {
- errorString << ", ";
- }
- errorString << "[";
- for (unsigned int i = 0; i < numberOfDimensions; ++i)
- {
- errorString << indices[i];
- if (i != numberOfDimensions - 1)
- {
- errorString << ",";
- }
- }
- errorString << "]";
-
- errorString << " (" << +actualData[index] << " != " << +expectedData[index] << ")";
- }
- }
-
- ++indices[numberOfDimensions - 1];
- for (unsigned int i=numberOfDimensions-1; i>0; i--)
- {
- if (indices[i] == actualShape[i])
- {
- indices[i] = 0;
- ++indices[i - 1];
- }
- }
- if (indices[0] == actualShape[0])
- {
- break;
- }
-
- index++;
- }
-
- armnn::PredicateResult comparisonResult(true);
- if (numFailedElements > 0)
- {
- comparisonResult.SetResult(false);
- comparisonResult.Message() << numFailedElements << " different values at: ";
- if (numFailedElements > maxReportedDifferences)
- {
- errorString << ", ... (and " << (numFailedElements - maxReportedDifferences) << " other differences)";
- }
- comparisonResult.Message() << errorString.str();
- }
-
- return comparisonResult;
-}
-
-template <typename T>
-std::vector<T> MakeRandomTensor(const armnn::TensorInfo& tensorInfo,
- unsigned int seed,
- float min = -10.0f,
- float max = 10.0f)
-{
- std::mt19937 gen(seed);
- std::uniform_real_distribution<float> dist(min, max);
-
- std::vector<float> init(tensorInfo.GetNumElements());
- for (unsigned int i = 0; i < init.size(); i++)
- {
- init[i] = dist(gen);
- }
-
- const float qScale = tensorInfo.GetQuantizationScale();
- const int32_t qOffset = tensorInfo.GetQuantizationOffset();
-
- return armnnUtils::QuantizedVector<T>(init, qScale, qOffset);
-}
diff --git a/src/armnnTestUtils/UnitTests.hpp b/src/armnnTestUtils/UnitTests.hpp
index 82b87a1cee..f560720c9b 100644
--- a/src/armnnTestUtils/UnitTests.hpp
+++ b/src/armnnTestUtils/UnitTests.hpp
@@ -4,7 +4,7 @@
//
#pragma once
-#include "TensorHelpers.hpp"
+#include "armnnTestUtils/TensorHelpers.hpp"
#include <armnn/Logging.hpp>
#include <armnn/Utils.hpp>
generated by cgit v1.2.1 (git 2.23.0) at 2024-05-23 15:20:21 +0000