79e6b0e0f6
As of today <zephyr/zephyr.h> is 100% equivalent to <zephyr/kernel.h>. This patch proposes to then include <zephyr/kernel.h> instead of <zephyr/zephyr.h> since it is more clear that you are including the Kernel APIs and (probably) nothing else. <zephyr/zephyr.h> sounds like a catch-all header that may be confusing. Most applications need to include a bunch of other things to compile, e.g. driver headers or subsystem headers like BT, logging, etc. The idea of a catch-all header in Zephyr is probably not feasible anyway. Reason is that Zephyr is not a library, like it could be for example `libpython`. Zephyr provides many utilities nowadays: a kernel, drivers, subsystems, etc and things will likely grow. A catch-all header would be massive, difficult to keep up-to-date. It is also likely that an application will only build a small subset. Note that subsystem-level headers may use a catch-all approach to make things easier, though. NOTE: This patch is **NOT** removing the header, just removing its usage in-tree. I'd advocate for its deprecation (add a #warning on it), but I understand many people will have concerns. Signed-off-by: Gerard Marull-Paretas <gerard.marull@nordicsemi.no>
84 lines
2.3 KiB
C
84 lines
2.3 KiB
C
/*
|
|
* 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 <zephyr/ztest.h>
|
|
#include <zephyr/kernel.h>
|
|
#include <stdlib.h>
|
|
#include <arm_math.h>
|
|
#include "../../common/test_common.h"
|
|
|
|
#include "f32.pat"
|
|
|
|
#define REL_ERROR_THRESH (5.0e-6)
|
|
|
|
ZTEST_SUITE(bayes_f32, NULL, NULL, NULL, NULL, NULL);
|
|
|
|
ZTEST(bayes_f32, test_gaussian_naive_bayes_predict_f32)
|
|
{
|
|
arm_gaussian_naive_bayes_instance_f32 inst;
|
|
|
|
size_t index;
|
|
const uint16_t pattern_count = in_dims[0];
|
|
const uint16_t class_count = in_dims[1];
|
|
const uint16_t vec_dims = in_dims[2];
|
|
|
|
const float32_t *params = (const float32_t *)in_param;
|
|
const float32_t *input = (const float32_t *)in_val;
|
|
float32_t *output_probs_buf, *output_probs;
|
|
uint16_t *output_preds_buf, *output_preds;
|
|
float32_t *temp;
|
|
|
|
/* Initialise instance */
|
|
inst.vectorDimension = vec_dims;
|
|
inst.numberOfClasses = class_count;
|
|
inst.theta = params;
|
|
inst.sigma = params + (class_count * vec_dims);
|
|
inst.classPriors = params + (2 * class_count * vec_dims);
|
|
inst.epsilon = params[class_count + (2 * class_count * vec_dims)];
|
|
|
|
/* Allocate output buffers */
|
|
output_probs_buf =
|
|
malloc(pattern_count * class_count * sizeof(float32_t));
|
|
zassert_not_null(output_probs_buf, ASSERT_MSG_BUFFER_ALLOC_FAILED);
|
|
|
|
output_preds_buf =
|
|
malloc(pattern_count * sizeof(uint16_t));
|
|
zassert_not_null(output_preds_buf, ASSERT_MSG_BUFFER_ALLOC_FAILED);
|
|
|
|
output_probs = output_probs_buf;
|
|
output_preds = output_preds_buf;
|
|
|
|
temp = malloc(pattern_count * class_count * sizeof(float32_t));
|
|
zassert_not_null(temp, ASSERT_MSG_BUFFER_ALLOC_FAILED);
|
|
|
|
/* Enumerate patterns */
|
|
for (index = 0; index < pattern_count; index++) {
|
|
/* Run test function */
|
|
*output_preds =
|
|
arm_gaussian_naive_bayes_predict_f32(
|
|
&inst, input, output_probs, temp);
|
|
|
|
/* Increment pointers */
|
|
input += vec_dims;
|
|
output_probs += class_count;
|
|
output_preds++;
|
|
}
|
|
|
|
/* Validate output */
|
|
zassert_true(
|
|
test_rel_error_f32(pattern_count, output_probs_buf,
|
|
(float32_t *)ref_prob, REL_ERROR_THRESH),
|
|
ASSERT_MSG_REL_ERROR_LIMIT_EXCEED);
|
|
|
|
zassert_true(
|
|
test_equal_q15(pattern_count, output_preds_buf, ref_pred),
|
|
ASSERT_MSG_INCORRECT_COMP_RESULT);
|
|
|
|
/* Free output buffers */
|
|
free(output_probs_buf);
|
|
free(output_preds_buf);
|
|
}
|