From 719a16b65e5d70264fae8320323bbf1bad24d206 Mon Sep 17 00:00:00 2001
From: Artemis Rosman <73006620+rozukke@users.noreply.github.com>
Date: Sun, 7 Jul 2024 23:37:18 +1000
Subject: [PATCH] Fix padding issue

---
 src/main.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main.c b/src/main.c
index a34da42..909586b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -34,8 +34,9 @@ void propagate_fwd(const matrix* weights, const float* inputs, float* results, c
 // Minumum number of alligned_alloc without breaking things.
 // This code f***ing sucks but its fast so uhhhh
 u8 infer_reuse_layers_thread(vector* input, matrix** weights, vector** biases) {
-    float out0[98] __attribute__((aligned(SIMD_ALIGN))) = {0};
-    float out1[65] __attribute__((aligned(SIMD_ALIGN))) = {0};
+    // Slightly larger than required for padding
+    float out0[104] __attribute__((aligned(SIMD_ALIGN))) = {0};
+    float out1[72] __attribute__((aligned(SIMD_ALIGN))) = {0};
 
     propagate_fwd(weights[0], input->data, out0, biases[0]);
     relu_inplace(out0, 98);