[systemml] branch master updated: [SYSTEMDS-411]Fix missing cache size update.

[ar...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

arnabp20 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/systemml.git


The following commit(s) were added to refs/heads/master by this push:
     new 7bf3517  [SYSTEMDS-411]Fix missing cache size update.
7bf3517 is described below

commit 7bf3517b11ad4fc303b97e952a2acbc6002909c5
Author: arnabp <ar...@tugraz.at>
AuthorDate: Mon Jun 1 23:15:54 2020 +0200

    [SYSTEMDS-411]Fix missing cache size update.
---
 .../java/org/apache/sysds/runtime/lineage/LineageCacheEviction.java   | 4 ++++
 1 file changed, 4 insertions(+)

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 @@ public class LineageCacheEviction
 				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 @@ public class LineageCacheEviction
 				h.setValue(mb);
 				h = h._nextEntry;
 			}
+			// Increase cachesize once for all the entries.
+			updateSize(e.getSize(), true);
 		}
 
 		// Adjust disk reading speed