fix test suite build failure

apache · Aug 21, 2018 · 3ec52f8 · 3ec52f8
1 parent 8bde467
commit 3ec52f8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/core/src/test/java/org/apache/spark/memory/TestMemoryConsumer.java b/core/src/test/java/org/apache/spark/memory/TestMemoryConsumer.java
@@ -19,6 +19,8 @@
 
 import java.io.IOException;
 
+import org.apache.spark.unsafe.memory.MemoryBlock;
+
 public class TestMemoryConsumer extends MemoryConsumer {
   public TestMemoryConsumer(TaskMemoryManager memoryManager, MemoryMode mode) {
     super(memoryManager, 1024L, mode);
@@ -43,6 +45,11 @@ void free(long size) {
     used -= size;
     taskMemoryManager.releaseExecutionMemory(size, this);
   }
+
+  public void freePage(MemoryBlock page) {
+    used -= page.size();
+    taskMemoryManager.freePage(page, this);
+  }
 }