From 60766967a494942a2b251046303fe3222df57262 Mon Sep 17 00:00:00 2001
From: Yury <rip95_95@mail.ru>
Date: Mon, 21 Oct 2024 12:27:46 +0300
Subject: [PATCH] Fix memory leak in quantize_i2_s

---
 src/ggml-bitnet-mad.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/ggml-bitnet-mad.cpp b/src/ggml-bitnet-mad.cpp
index f75e6ca..eeca82b 100644
--- a/src/ggml-bitnet-mad.cpp
+++ b/src/ggml-bitnet-mad.cpp
@@ -85,6 +85,8 @@ size_t quantize_i2_s(const float * src, void * dst, int64_t nrow, int64_t n_per_
     float* scale_ptr = (float*)((char*)i2_weight + n / 4);
     scale_ptr[0] = i2_scale;
 
+    free(q8);
+
     // 32B for alignment
     return nrow * row_size / 4 + 32;
 }