diff options
author | Jack Frankland <jack.frankland@arm.com> | 2023-09-20 09:08:34 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-10-03 21:48:32 +0000 |
commit | 12ee1a79374b451602784fd6dc8f63886bf2a997 (patch) | |
tree | b01d4ee8c0ee1c1d276d7b975bfaa78dc3ceee66 /reference_model/src/verify/verify_reduce_product.cc | |
parent | e2b5e87804e158cb3e5d06a131c317b3890b87b3 (diff) | |
download | reference_model-12ee1a79374b451602784fd6dc8f63886bf2a997.tar.gz |
Add reduce product verifier
* Add verifiers to validate the result of a reduce produce operation.
* Add test cases for the new validator.
Change-Id: I666d1a67f498e7893e0f224bc5408a4134f2ef6c
Signed-off-by: Jack Frankland <jack.frankland@arm.com>
Diffstat (limited to 'reference_model/src/verify/verify_reduce_product.cc')
-rw-r--r-- | reference_model/src/verify/verify_reduce_product.cc | 88 |
1 files changed, 88 insertions, 0 deletions
diff --git a/reference_model/src/verify/verify_reduce_product.cc b/reference_model/src/verify/verify_reduce_product.cc new file mode 100644 index 0000000..d233b0f --- /dev/null +++ b/reference_model/src/verify/verify_reduce_product.cc @@ -0,0 +1,88 @@ + +// Copyright (c) 2023, ARM Limited. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include <cmath> +#include <vector> + +#include "verifiers.h" +#include "verify/verify_utils.h" + +namespace +{ + +auto calculateError(uint64_t M, uint64_t N) +{ + return std::pow(1 + std::pow(2, -static_cast<int64_t>(M) - 1), N) - 1; +} + +template <typename FP> +auto calculateTolerance(uint64_t M, uint64_t N, FP value) +{ + return std::abs(value) * calculateError(M, N); +} +} // namespace + +namespace TosaReference +{ + +bool verifyReduceProduct(const CTensor* referenceTensor, const CTensor* implementationTensor, uint64_t m, uint64_t n) +{ + // Validate that tensors are provided + TOSA_REF_REQUIRE(referenceTensor != nullptr, "reference tensor is missing"); + TOSA_REF_REQUIRE(implementationTensor != nullptr, "implementation tensor is missing"); + + // Get number of elements + const auto elementCount = + numElements(std::vector<int32_t>(referenceTensor->shape, referenceTensor->shape + referenceTensor->num_dims)); + TOSA_REF_REQUIRE(elementCount > 0, "invalid shape for reference tensor"); + + switch (implementationTensor->data_type) + { + case tosa_datatype_fp32_t: { + const auto* refData = reinterpret_cast<const float*>(referenceTensor->data); + TOSA_REF_REQUIRE(refData != nullptr, "missing data for reference"); + + const auto* impData = reinterpret_cast<const float*>(implementationTensor->data); + TOSA_REF_REQUIRE(impData != nullptr, "missing data for implementation"); + + return std::equal(refData, std::next(refData, elementCount), impData, std::next(impData, elementCount), + [m, n](const auto& referenceValue, const auto& implementationValue) { + // Result overflows must be set to zero of the correct sign. + if (std::isinf(implementationValue)) + { + return implementationValue == referenceValue; + } + + // Result underflows must be set to a zero of the correct sign. + if (implementationValue == 0.f || implementationValue == -0.f) + { + return implementationValue == referenceValue; + } + + // Otherwise we are in the normal range. + const auto absoulteError = (referenceValue < implementationValue) + ? implementationValue - referenceValue + : referenceValue - implementationValue; + return absoulteError <= calculateTolerance(m, n, implementationValue); + }); + } + default: + WARNING("tosa verifier: data-type not supported."); + break; + } + + return false; +} +} // namespace TosaReference |