Fix padding issue

kachi-group · Jul 7, 2024 · 719a16b · 719a16b
1 parent 6e8df3f
commit 719a16b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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);