From 1f3db496cc3b0d2de4243aa6ee565bf7860442d6 Mon Sep 17 00:00:00 2001
From: Ed Addario <eaddario@hotmail.com>
Date: Mon, 17 Nov 2025 13:36:28 +0000
Subject: [PATCH] Calculate layer_sum only for legacy

---
 tools/imatrix/imatrix.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tools/imatrix/imatrix.cpp b/tools/imatrix/imatrix.cpp
index 407b78fece..34d222e12c 100644
--- a/tools/imatrix/imatrix.cpp
+++ b/tools/imatrix/imatrix.cpp
@@ -1455,12 +1455,13 @@ static bool show_statistics(const common_params & params) {
 
         const float zd = (float)tstat.elements * tstat.zd_score;
         if (ls.find(blk) != ls.end()) {
-            ls[blk].layer_sum += tstat.sum_values;
+            if (legacy) { ls[blk].layer_sum += tstat.sum_values; }
             ls[blk].layer_zd += zd;
             ls[blk].n += tstat.elements;
         } else {
             layer_stats temp_ls;
-            temp_ls.layer_sum = tstat.sum_values;
+            if (legacy) { temp_ls.layer_sum = tstat.sum_values; }
+            else { temp_ls.layer_sum = 0.0f; }
             temp_ls.layer_zd = zd;
             temp_ls.n = tstat.elements;
             ls[blk] = temp_ls;