From 8397fb1512d088e479acb23503ad1ffc34e51a96 Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <88108002+VinciGit00@users.noreply.github.com>
Date: Tue, 9 Apr 2024 20:14:18 +0200
Subject: [PATCH] Free the memory in layernorm.c

---
 doc/layernorm/layernorm.c | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/doc/layernorm/layernorm.c b/doc/layernorm/layernorm.c
index e6067bbc4..09b1782dc 100644
--- a/doc/layernorm/layernorm.c
+++ b/doc/layernorm/layernorm.c
@@ -161,6 +161,15 @@ int main() {
     check_tensor(c_dw, dw, C, "dw");
     check_tensor(c_db, db, C, "db");
 
-    // TODO free stuff
+    free(x);
+    free(w);
+    free(b);
+    free(out);
+    free(mean);
+    free(rstd);
+    free(dout);
+    free(dx);
+    free(dw);
+    free(db);
     return 0;
-}
\ No newline at end of file
+}