Free the memory in layernorm.c

suisui-koubou · Apr 9, 2024 · 8397fb1 · 8397fb1
1 parent 03f37cf
commit 8397fb1
Showing 1 changed file with 11 additions and 2 deletions.
diff --git 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;
-}
+}