diff options
| author | joshua <joshua@joshuayun.com> | 2023-12-30 23:54:31 -0500 |
|---|---|---|
| committer | joshua <joshua@joshuayun.com> | 2023-12-30 23:54:31 -0500 |
| commit | 86608c6770cf08c138a2bdab5855072f64be09ef (patch) | |
| tree | 494a61b3ef37e76f9235a0d10f5c93d97290a35f /Drivers/CMSIS/NN/NN_Lib_Tests/nn_test/arm_nnexamples_nn_test.h | |
| download | sdr-software-master.tar.gz | |
Diffstat (limited to 'Drivers/CMSIS/NN/NN_Lib_Tests/nn_test/arm_nnexamples_nn_test.h')
| -rw-r--r-- | Drivers/CMSIS/NN/NN_Lib_Tests/nn_test/arm_nnexamples_nn_test.h | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/Drivers/CMSIS/NN/NN_Lib_Tests/nn_test/arm_nnexamples_nn_test.h b/Drivers/CMSIS/NN/NN_Lib_Tests/nn_test/arm_nnexamples_nn_test.h new file mode 100644 index 0000000..2e33988 --- /dev/null +++ b/Drivers/CMSIS/NN/NN_Lib_Tests/nn_test/arm_nnexamples_nn_test.h @@ -0,0 +1,78 @@ +#ifndef _MAIN_H_
+#define _MAIN_H_
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <math.h>
+
+#include "arm_math.h"
+
+#include "arm_nnfunctions.h"
+#include "ref_functions.h"
+
+extern int test_index;
+extern q7_t test_flags[50];
+
+void initialize_results_q7(q7_t * ref, q7_t * opt, int length)
+{
+ arm_fill_q7(0, ref, length);
+ arm_fill_q7(37, opt, length);
+}
+
+void initialize_results_q15(q15_t * ref, q15_t * opt, int length)
+{
+ arm_fill_q15(0, ref, length);
+ arm_fill_q15(0x5F5, opt, length);
+}
+
+void verify_results_q7(q7_t * ref, q7_t * opt, int length)
+{
+
+ bool if_match = true;
+
+ for (int i = 0; i < length; i++)
+ {
+ if (ref[i] != opt[i])
+ {
+ printf("Output mismatch at %d, expected %d, actual %d\r\n", i, ref[i], opt[i]);
+
+ if_match = false;
+ }
+ }
+
+ if (if_match == true)
+ {
+ printf("Outputs match.\r\n\r\n");
+ test_flags[test_index++] = 0;
+ } else {
+ test_flags[test_index++] = 1;
+ }
+
+}
+
+void verify_results_q15(q15_t * ref, q15_t * opt, int length)
+{
+
+ bool if_match = true;
+
+ for (int i = 0; i < length; i++)
+ {
+ if (ref[i] != opt[i])
+ {
+ printf("Output mismatch at %d, expected %d, actual %d\r\n", i, ref[i], opt[i]);
+
+ if_match = false;
+ }
+ }
+
+ if (if_match == true)
+ {
+ printf("Outputs match.\r\n\r\n");
+ test_flags[test_index++] = 0;
+ } else {
+ test_flags[test_index++] = 1;
+ }
+
+}
+
+#endif
|