From fae646bc11bf0ae769374a23956c74f6bdd14e77 Mon Sep 17 00:00:00 2001
From: Artemis Rosman <73006620+rozukke@users.noreply.github.com>
Date: Thu, 4 Jul 2024 01:51:59 +1000
Subject: [PATCH] Fix format

---
 src/main.c   | 4 ++--
 src/matrix.c | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/main.c b/src/main.c
index 3163ab8..211703e 100644
--- a/src/main.c
+++ b/src/main.c
@@ -208,8 +208,8 @@ int main(int argc, char* argv[]) {
         for (int i = 0; i < input_count; i++) {
             // printf("Thread %d: Processing input %d\n", omp_get_thread_num(), i);
 
-            vector* input = new_vec_aligned(TSIZE_ALGN_BYTES / sizeof(f32));
-            memcpy(input->data, (f32*)&tensors[TSIZE_ALGN_BYTES / sizeof(f32) * i], TSIZE_ALGN_BYTES);
+            vector* input = new_vec_aligned(TENSOR_SIZE);
+            memcpy(input->data, (f32*)&tensors[TSIZE_ALGN_BYTES / sizeof(f32) * i], TENSOR_SIZE * sizeof(f32));
 
 #pragma omp for
             for (int j = 0; j < iter_per_in - 1; j++) {
diff --git a/src/matrix.c b/src/matrix.c
index 1624f82..3287146 100644
--- a/src/matrix.c
+++ b/src/matrix.c
@@ -107,7 +107,6 @@ u8 argmax(f32* in, int len) {
     return idx;
 }
 
-
 void transpose_mat_inplace(matrix* in) {
     int cols_before = in->cols;
     int rows_before = in->rows;