riot-ci/RIOT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tests/pkg_utensor: fix alignment bug

Browse Source
This commit is contained in:
Marian Buschsieweke 2021-11-12 12:07:20 +01:00
parent 1193785a11
commit 3a07baf8e6
No known key found for this signature in database
GPG Key ID: CB8E3238CE715A94
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/pkg_utensor/main.cpp
View File

@ -18,9 +18,11 @@
* @} * @}
*/ */
#include <stdalign.h>
#include <stdio.h> #include <stdio.h>
#include <inttypes.h> #include <inttypes.h>
alignas(float)
#include "blob/digit.h" //contains a sample taken from the MNIST test set #include "blob/digit.h" //contains a sample taken from the MNIST test set
#include "deep_mlp.hpp" //generated model file #include "deep_mlp.hpp" //generated model file
@ -33,8 +35,12 @@ int main()
// create the context class, the stage where inferences take place // create the context class, the stage where inferences take place
Context ctx; Context ctx;
// because we used alignas(float), we can rest assured that silencing
// -Wcast-align with an intermediate cast to uintptr_t is fine
float *digit_as_float = (float *)(uintptr_t)digit;
// wrap the input digit in a tensor class // wrap the input digit in a tensor class
auto input_x = new WrappedRamTensor<float>({1, digit_len >> 2}, (float *)digit); auto input_x = new WrappedRamTensor<float>({1, digit_len >> 2}, digit_as_float);
// pass ownership of the tensor to the context // pass ownership of the tensor to the context
get_deep_mlp_ctx(ctx, input_x); get_deep_mlp_ctx(ctx, input_x);