[SYSTEMDS-411]Fix missing cache size update.
diff --git a/src/main/java/org/apache/sysds/runtime/lineage/LineageCacheEviction.java b/src/main/java/org/apache/sysds/runtime/lineage/LineageCacheEviction.java
index b83a78b..2139689 100644
--- a/src/main/java/org/apache/sysds/runtime/lineage/LineageCacheEviction.java
+++ b/src/main/java/org/apache/sysds/runtime/lineage/LineageCacheEviction.java
@@ -137,6 +137,8 @@
h.setCacheStatus(LineageCacheStatus.SPILLED);
h = h._nextEntry;
}
+ // Reduce cachesize once for all the entries.
+ updateSize(e.getSize(), false);
// Keep them in cache.
return;
}
@@ -371,6 +373,8 @@
h.setValue(mb);
h = h._nextEntry;
}
+ // Increase cachesize once for all the entries.
+ updateSize(e.getSize(), true);
}
// Adjust disk reading speed
