michaelh/zephyr
1
0
Fork 0
Code Releases Activity

tests: lib: cmsis_dsp: common: Add f16 test routines

Browse source 
This commit adds the common test and validation routines for the half-
precision floating point data.

Signed-off-by: Stephanos Ioannidis <root@stephanos.io>
This commit is contained in:
Stephanos Ioannidis 2021-08-18 21:50:52 +09:00 • committed by Carles Cufí
commit 0456de7bea
2 changed files with 130 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

38
tests/lib/cmsis_dsp/common/math_helper.h
View file

@ -39,7 +39,10 @@
#ifndef MATH_HELPER_H
#define MATH_HELPER_H
#include "arm_math.h"
#include <arm_math.h>
#ifdef CONFIG_CMSIS_DSP_FLOAT16
#include <arm_math_f16.h>
#endif
/**
* @brief Calculation of SNR
@ -131,6 +134,39 @@ static inline float arm_snr_f32(const float *pRef, const float *pTest,
return SNR;
}
#ifdef CONFIG_CMSIS_DSP_FLOAT16
static inline float arm_snr_f16(const float16_t *pRef, const float16_t *pTest,
uint32_t buffSize)
{
float EnergySignal = 0.0, EnergyError = 0.0;
uint32_t i;
float SNR;
for (i = 0; i < buffSize; i++) {
/* Checking for a NAN value in pRef array */
IFNANRETURNZERO((float)pRef[i]);
/* Checking for a NAN value in pTest array */
IFNANRETURNZERO((float)pTest[i]);
EnergySignal += pRef[i] * pRef[i];
EnergyError += (pRef[i] - pTest[i]) * (pRef[i] - pTest[i]);
}
/* Checking for a NAN value in EnergyError */
IFNANRETURNZERO(EnergyError);
SNR = 10 * log10f(EnergySignal / EnergyError);
/* Checking for a NAN value in SNR */
IFNANRETURNZERO(SNR);
IFINFINITERETURN(SNR, 100000.0);
return SNR;
}
#endif /* CONFIG_CMSIS_DSP_FLOAT16 */
static inline float arm_snr_q63(const q63_t *pRef, const q63_t *pTest,
uint32_t buffSize)
{

95
tests/lib/cmsis_dsp/common/test_common.h
View file

@ -1,6 +1,6 @@
/*
* Copyright (c) 2020 Stephanos Ioannidis <root@stephanos.io>
* Copyright (C) 2010-2020 ARM Limited or its affiliates. All rights reserved.
* Copyright (c) 2021 Stephanos Ioannidis <root@stephanos.io>
* Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
*
* SPDX-License-Identifier: Apache-2.0
*/
@ -12,6 +12,9 @@
#include <zephyr.h>
#include <stdlib.h>
#include <arm_math.h>
#ifdef CONFIG_CMSIS_DSP_FLOAT16
#include <arm_math_f16.h>
#endif
#include "math_helper.h"
@ -96,6 +99,22 @@ static inline bool test_equal_f32(
return true;
}
#ifdef CONFIG_CMSIS_DSP_FLOAT16
static inline bool test_equal_f16(
size_t length, const float16_t *a, const float16_t *b)
{
size_t index;
for (index = 0; index < length; index++) {
if (a[index] != b[index]) {
return false;
}
}
return true;
}
#endif /* CONFIG_CMSIS_DSP_FLOAT16 */
static inline bool test_equal_q63(
size_t length, const q63_t *a, const q63_t *b)
{
@ -182,6 +201,23 @@ static inline bool test_near_equal_f32(
return true;
}
#ifdef CONFIG_CMSIS_DSP_FLOAT16
static inline bool test_near_equal_f16(
size_t length, const float16_t *a, const float16_t *b,
float16_t threshold)
{
size_t index;
for (index = 0; index < length; index++) {
if (fabs(a[index] - b[index]) > threshold) {
return false;
}
}
return true;
}
#endif /* CONFIG_CMSIS_DSP_FLOAT16 */
static inline bool test_near_equal_q63(
size_t length, const q63_t *a, const q63_t *b, q63_t threshold)
{
@ -284,6 +320,31 @@ static inline bool test_rel_error_f32(
return true;
}
#ifdef CONFIG_CMSIS_DSP_FLOAT16
static inline bool test_rel_error_f16(
size_t length, const float16_t *a, const float16_t *b,
float16_t threshold)
{
size_t index;
float32_t rel, delta, average;
for (index = 0; index < length; index++) {
delta = fabs(a[index] - b[index]);
average = (fabs(a[index]) + fabs(b[index])) / 2.0f;
if (average != 0) {
rel = delta / average;
if (rel > threshold) {
return false;
}
}
}
return true;
}
#endif /* CONFIG_CMSIS_DSP_FLOAT16 */
static inline bool test_close_error_f64(
size_t length, const float64_t *ref, const float64_t *val,
float64_t abs_threshold, float64_t rel_threshold)
@ -316,6 +377,24 @@ static inline bool test_close_error_f32(
return true;
}
#ifdef CONFIG_CMSIS_DSP_FLOAT16
static inline bool test_close_error_f16(
size_t length, const float16_t *ref, const float16_t *val,
float32_t abs_threshold, float32_t rel_threshold)
{
size_t index;
for (index = 0; index < length; index++) {
if (fabs(val[index] - ref[index]) >
(abs_threshold + rel_threshold * fabs(ref[index]))) {
return false;
}
}
return true;
}
#endif /* CONFIG_CMSIS_DSP_FLOAT16 */
static inline bool test_snr_error_f64(
size_t length, const float64_t *a, const float64_t *b,
float64_t threshold)
@ -336,6 +415,18 @@ static inline bool test_snr_error_f32(
return (snr >= threshold);
}
#ifdef CONFIG_CMSIS_DSP_FLOAT16
static inline bool test_snr_error_f16(
size_t length, const float16_t *a, const float16_t *b,
float32_t threshold)
{
float32_t snr;
snr = arm_snr_f16(a, b, length);
return (snr >= threshold);
}
#endif /* CONFIG_CMSIS_DSP_FLOAT16 */
static inline bool test_snr_error_q63(
size_t length, const q63_t *a, const q63_t *b, float32_t threshold)
{