tests: lib: cmsis_dsp: transform: Add RF16 tests for 1.9.0

This commit adds the transform RF16 test patterns and implementations
for the CMSIS-DSP 1.9.0.

Signed-off-by: Stephanos Ioannidis <root@stephanos.io>

tests/lib/cmsis_dsp/transform/CMakeLists.txt

@@ -28,6 +28,11 @@ target_sources_ifdef(
   app PRIVATE src/rq31.c
   )
 
+target_sources_ifdef(
+  CONFIG_CMSIS_DSP_TEST_TRANSFORM_RF16
+  app PRIVATE src/rf16.c
+  )
+
 target_sources_ifdef(
   CONFIG_CMSIS_DSP_TEST_TRANSFORM_CF32
   app PRIVATE src/cf32.c

tests/lib/cmsis_dsp/transform/Kconfig

@@ -1,4 +1,4 @@
-# Copyright (c) 2019 Stephanos Ioannidis <root@stephanos.io>
+# Copyright (c) 2021 Stephanos Ioannidis <root@stephanos.io>
 # SPDX-License-Identifier: Apache-2.0
 
 config CMSIS_DSP_TEST_TRANSFORM_CQ15
@@ -13,6 +13,10 @@ config CMSIS_DSP_TEST_TRANSFORM_CQ31
 config CMSIS_DSP_TEST_TRANSFORM_RQ31
 	bool "Test: Transform RQ31"
 
+config CMSIS_DSP_TEST_TRANSFORM_RF16
+	bool "Test: Transform RF16"
+	depends on CMSIS_DSP_FLOAT16
+
 config CMSIS_DSP_TEST_TRANSFORM_CF32
 	bool "Test: Transform CF32"
 

tests/lib/cmsis_dsp/transform/prj.conf

@@ -8,6 +8,7 @@ CONFIG_CMSIS_DSP_TEST_TRANSFORM_CQ15=y
 CONFIG_CMSIS_DSP_TEST_TRANSFORM_RQ15=y
 CONFIG_CMSIS_DSP_TEST_TRANSFORM_CQ31=y
 CONFIG_CMSIS_DSP_TEST_TRANSFORM_RQ31=y
+CONFIG_CMSIS_DSP_TEST_TRANSFORM_RF16=y
 CONFIG_CMSIS_DSP_TEST_TRANSFORM_CF32=y
 CONFIG_CMSIS_DSP_TEST_TRANSFORM_RF32=y
 CONFIG_CMSIS_DSP_TEST_TRANSFORM_CF64=y

tests/lib/cmsis_dsp/transform/src/main.c

@@ -11,6 +11,7 @@ extern void test_transform_cq15(void);
 extern void test_transform_rq15(void);
 extern void test_transform_cq31(void);
 extern void test_transform_rq31(void);
+extern void test_transform_rf16(void);
 extern void test_transform_cf32(void);
 extern void test_transform_rf32(void);
 extern void test_transform_cf64(void);
@@ -30,6 +31,9 @@ void test_main(void)
 #ifdef CONFIG_CMSIS_DSP_TEST_TRANSFORM_RQ31
 	test_transform_rq31();
 #endif
+#ifdef CONFIG_CMSIS_DSP_TEST_TRANSFORM_RF16
+	test_transform_rf16();
+#endif
 #ifdef CONFIG_CMSIS_DSP_TEST_TRANSFORM_CF32
 	test_transform_cf32();
 #endif

tests/lib/cmsis_dsp/transform/src/rf16.c

@@ -0,0 +1,199 @@
+/*
+ * 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
+ */
+
+#include <ztest.h>
+#include <zephyr.h>
+#include <stdlib.h>
+#include <arm_math_f16.h>
+#include <arm_const_structs.h>
+#include "../../common/test_common.h"
+
+#include "rf16.pat"
+
+#define SNR_ERROR_THRESH	((float32_t)58)
+
+static void test_arm_rfft_f16_real_backend(
+	bool inverse, const uint16_t *input, const uint16_t *ref,
+	size_t length)
+{
+	arm_rfft_fast_instance_f16 inst;
+	float16_t *output, *scratch;
+
+	/* Initialise instance */
+	arm_rfft_fast_init_f16(&inst, length);
+
+	/* Allocate output buffer */
+	output = malloc(length * sizeof(float16_t));
+	zassert_not_null(output, ASSERT_MSG_BUFFER_ALLOC_FAILED);
+
+	scratch = calloc(length + 2, sizeof(float16_t)); /* see #24701 */
+	zassert_not_null(scratch, ASSERT_MSG_BUFFER_ALLOC_FAILED);
+
+	/* Load data in place */
+	memcpy(scratch, input, length * sizeof(float16_t));
+
+	/* Run test function */
+	arm_rfft_fast_f16(&inst, scratch, output, inverse);
+
+	/* Validate output */
+	zassert_true(
+		test_snr_error_f16(length, output, (float16_t *)ref,
+			SNR_ERROR_THRESH),
+		ASSERT_MSG_SNR_LIMIT_EXCEED);
+
+	/* Free output buffer */
+	free(output);
+	free(scratch);
+}
+
+static void test_arm_rfft_f16_real(
+	const uint16_t *input, const uint16_t *ref, size_t length)
+{
+	test_arm_rfft_f16_real_backend(false, input, ref, length);
+}
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_32,
+	in_rfft_noisy_32, ref_rfft_noisy_32, 32);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_64,
+	in_rfft_noisy_64, ref_rfft_noisy_64, 64);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_128,
+	in_rfft_noisy_128, ref_rfft_noisy_128, 128);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_256,
+	in_rfft_noisy_256, ref_rfft_noisy_256, 256);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_512,
+	in_rfft_noisy_512, ref_rfft_noisy_512, 512);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_1024,
+	in_rfft_noisy_1024, ref_rfft_noisy_1024, 1024);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_2048,
+	in_rfft_noisy_2048, ref_rfft_noisy_2048, 2048);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, noisy_4096,
+	in_rfft_noisy_4096, ref_rfft_noisy_4096, 4096);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_32,
+	in_rfft_step_32, ref_rfft_step_32, 32);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_64,
+	in_rfft_step_64, ref_rfft_step_64, 64);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_128,
+	in_rfft_step_128, ref_rfft_step_128, 128);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_256,
+	in_rfft_step_256, ref_rfft_step_256, 256);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_512,
+	in_rfft_step_512, ref_rfft_step_512, 512);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_1024,
+	in_rfft_step_1024, ref_rfft_step_1024, 1024);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_2048,
+	in_rfft_step_2048, ref_rfft_step_2048, 2048);
+
+DEFINE_TEST_VARIANT3(arm_rfft_f16_real, step_4096,
+	in_rfft_step_4096, ref_rfft_step_4096, 4096);
+
+static void test_arm_rifft_f16_real(
+	const uint16_t *input, const uint16_t *ref, size_t length)
+{
+	test_arm_rfft_f16_real_backend(true, input, ref, length);
+}
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_32,
+	in_rifft_noisy_32, in_rfft_noisy_32, 32);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_64,
+	in_rifft_noisy_64, in_rfft_noisy_64, 64);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_128,
+	in_rifft_noisy_128, in_rfft_noisy_128, 128);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_256,
+	in_rifft_noisy_256, in_rfft_noisy_256, 256);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_512,
+	in_rifft_noisy_512, in_rfft_noisy_512, 512);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_1024,
+	in_rifft_noisy_1024, in_rfft_noisy_1024, 1024);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_2048,
+	in_rifft_noisy_2048, in_rfft_noisy_2048, 2048);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, noisy_4096,
+	in_rifft_noisy_4096, in_rfft_noisy_4096, 4096);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_32,
+	in_rifft_step_32, in_rfft_step_32, 32);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_64,
+	in_rifft_step_64, in_rfft_step_64, 64);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_128,
+	in_rifft_step_128, in_rfft_step_128, 128);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_256,
+	in_rifft_step_256, in_rfft_step_256, 256);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_512,
+	in_rifft_step_512, in_rfft_step_512, 512);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_1024,
+	in_rifft_step_1024, in_rfft_step_1024, 1024);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_2048,
+	in_rifft_step_2048, in_rfft_step_2048, 2048);
+
+DEFINE_TEST_VARIANT3(arm_rifft_f16_real, step_4096,
+	in_rifft_step_4096, in_rfft_step_4096, 4096);
+
+void test_transform_rf16(void)
+{
+	ztest_test_suite(transform_rf16,
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_32),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_32),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_64),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_64),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_128),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_128),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_256),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_256),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_512),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_512),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_1024),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_1024),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_2048),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_2048),
+		ztest_unit_test(test_arm_rfft_f16_real_noisy_4096),
+		ztest_unit_test(test_arm_rifft_f16_real_noisy_4096),
+		ztest_unit_test(test_arm_rfft_f16_real_step_32),
+		ztest_unit_test(test_arm_rifft_f16_real_step_32),
+		ztest_unit_test(test_arm_rfft_f16_real_step_64),
+		ztest_unit_test(test_arm_rifft_f16_real_step_64),
+		ztest_unit_test(test_arm_rfft_f16_real_step_128),
+		ztest_unit_test(test_arm_rifft_f16_real_step_128),
+		ztest_unit_test(test_arm_rfft_f16_real_step_256),
+		ztest_unit_test(test_arm_rifft_f16_real_step_256),
+		ztest_unit_test(test_arm_rfft_f16_real_step_512),
+		ztest_unit_test(test_arm_rifft_f16_real_step_512),
+		ztest_unit_test(test_arm_rfft_f16_real_step_1024),
+		ztest_unit_test(test_arm_rifft_f16_real_step_1024),
+		ztest_unit_test(test_arm_rfft_f16_real_step_2048),
+		ztest_unit_test(test_arm_rifft_f16_real_step_2048),
+		ztest_unit_test(test_arm_rfft_f16_real_step_4096),
+		ztest_unit_test(test_arm_rifft_f16_real_step_4096)
+		);
+
+	ztest_run_test_suite(transform_rf16);
+}

tests/lib/cmsis_dsp/transform/testcase.yaml

@@ -99,6 +99,30 @@ tests:
     extra_configs:
       - CONFIG_CMSIS_DSP_TEST_TRANSFORM_RQ31=y
       - CONFIG_FPU=y
+  libraries.cmsis_dsp.transform.rf16:
+    filter: ((CONFIG_CPU_CORTEX_R or CONFIG_CPU_CORTEX_M) and TOOLCHAIN_HAS_NEWLIB == 1) or CONFIG_ARCH_POSIX
+    integration_platforms:
+      - frdm_k64f
+      - sam_e70_xplained
+      - mps2_an521
+      - native_posix
+    tags: cmsis_dsp
+    min_flash: 512
+    min_ram: 64
+    extra_args: CONF_FILE=prj_base.conf
+    extra_configs:
+      - CONFIG_CMSIS_DSP_TEST_TRANSFORM_RF16=y
+  libraries.cmsis_dsp.transform.rf16.fpu:
+    filter: ((CONFIG_CPU_CORTEX_R or CONFIG_CPU_CORTEX_M) and CONFIG_CPU_HAS_FPU and TOOLCHAIN_HAS_NEWLIB == 1) or CONFIG_ARCH_POSIX
+    integration_platforms:
+      - mps2_an521_remote
+    tags: cmsis_dsp fpu
+    min_flash: 512
+    min_ram: 64
+    extra_args: CONF_FILE=prj_base.conf
+    extra_configs:
+      - CONFIG_CMSIS_DSP_TEST_TRANSFORM_RF16=y
+      - CONFIG_FPU=y
   libraries.cmsis_dsp.transform.cf32:
     filter: ((CONFIG_CPU_CORTEX_R or CONFIG_CPU_CORTEX_M) and TOOLCHAIN_HAS_NEWLIB == 1) or CONFIG_ARCH_POSIX
     integration_platforms: