[CMSIS-NN] Fix memory alignment bug in CMSIS-NN demo (apache#11221)

* Updates convert_image.py to include memory alignment
shingjan · May 17, 2022 · f43d91f · f43d91f
1 parent 5adac43
commit f43d91f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apps/microtvm/cmsisnn/convert_image.py b/apps/microtvm/cmsisnn/convert_image.py
@@ -34,7 +34,7 @@ def create_header_file(name, tensor_name, tensor_data, output_path):
         header_file.write(
             "\n"
             + f"const size_t {tensor_name}_len = {tensor_data.size};\n"
-            + f'int8_t {tensor_name}[] = "'
+            + f'__attribute__((section(".data.tvm"), aligned(16))) int8_t {tensor_name}[] = "'
         )
 
         data_hexstr = tensor_data.tobytes().hex()