tests: lib: cmsis_dsp: bayes: Update F32 tests for 1.9.0
This commit updates the bayes F32 test implementations for the CMSIS-DSP 1.9.0. Signed-off-by: Stephanos Ioannidis <root@stephanos.io>
This commit is contained in:
parent
a2eb438ae0
commit
198c0845b9
1 changed files with 7 additions and 3 deletions
|
@ -1,6 +1,6 @@
|
||||||
/*
|
/*
|
||||||
* Copyright (c) 2020 Stephanos Ioannidis <root@stephanos.io>
|
* Copyright (c) 2021 Stephanos Ioannidis <root@stephanos.io>
|
||||||
* Copyright (C) 2010-2020 ARM Limited or its affiliates. All rights reserved.
|
* Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
|
||||||
*
|
*
|
||||||
* SPDX-License-Identifier: Apache-2.0
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
*/
|
*/
|
||||||
|
@ -28,6 +28,7 @@ void test_gaussian_naive_bayes_predict_f32(void)
|
||||||
const float32_t *input = (const float32_t *)in_val;
|
const float32_t *input = (const float32_t *)in_val;
|
||||||
float32_t *output_probs_buf, *output_probs;
|
float32_t *output_probs_buf, *output_probs;
|
||||||
uint16_t *output_preds_buf, *output_preds;
|
uint16_t *output_preds_buf, *output_preds;
|
||||||
|
float32_t *temp;
|
||||||
|
|
||||||
/* Initialise instance */
|
/* Initialise instance */
|
||||||
inst.vectorDimension = vec_dims;
|
inst.vectorDimension = vec_dims;
|
||||||
|
@ -49,12 +50,15 @@ void test_gaussian_naive_bayes_predict_f32(void)
|
||||||
output_probs = output_probs_buf;
|
output_probs = output_probs_buf;
|
||||||
output_preds = output_preds_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 */
|
/* Enumerate patterns */
|
||||||
for (index = 0; index < pattern_count; index++) {
|
for (index = 0; index < pattern_count; index++) {
|
||||||
/* Run test function */
|
/* Run test function */
|
||||||
*output_preds =
|
*output_preds =
|
||||||
arm_gaussian_naive_bayes_predict_f32(
|
arm_gaussian_naive_bayes_predict_f32(
|
||||||
&inst, input, output_probs);
|
&inst, input, output_probs, temp);
|
||||||
|
|
||||||
/* Increment pointers */
|
/* Increment pointers */
|
||||||
input += vec_dims;
|
input += vec_dims;
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue