correct counts on load

CISC · Jun 13, 2024 · fe21ef7 · fe21ef7
1 parent 4ad3eb2
commit fe21ef7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/imatrix/imatrix.cpp b/examples/imatrix/imatrix.cpp
@@ -343,8 +343,10 @@ bool IMatrixCollector::load_imatrix(const char * fname) {
 
         // Recreate the state as expected by save_imatrix(), and corerct for weighted sum.
         for (int i = 0; i < nval; i++) {
-            e.values[i] += tmp[i];
-            e.counts[i] += ncall;
+            if (tmp[i]) {
+                e.values[i] += tmp[i];
+                e.counts[i] += ncall;
+            }
         }
         e.ncall += ncall;