Fix compilation issue.

JoshRosen · JoshRosen · commit c3d0f6124576 · 2016-03-25T11:26:54.000-07:00
diff --git a/core/src/main/scala/org/apache/spark/storage/memory/MemoryStore.scala b/core/src/main/scala/org/apache/spark/storage/memory/MemoryStore.scala
@@ -354,7 +354,7 @@ private[spark] class MemoryStore(
       // Synchronize so that transfer is atomic
       memoryManager.synchronized {
         releaseUnrollMemoryForThisTask(unrollMemoryUsedByThisBlock)
-        val success = memoryManager.acquireStorageMemory(blockId, entry.size)
+        val success = memoryManager.acquireStorageMemory(blockId, entry.size, MemoryMode.ON_HEAP)
         assert(success, "transferring unroll memory to storage memory failed")
       }
       entries.synchronized {

Original file line number	Diff line number	Diff line change
`@@ -354,7 +354,7 @@ private[spark] class MemoryStore(`
`354`	`354`	`// Synchronize so that transfer is atomic`
`355`	`355`	`memoryManager.synchronized {`
`356`	`356`	`releaseUnrollMemoryForThisTask(unrollMemoryUsedByThisBlock)`
`357`		`- val success = memoryManager.acquireStorageMemory(blockId, entry.size)`
	`357`	`+ val success = memoryManager.acquireStorageMemory(blockId, entry.size, MemoryMode.ON_HEAP)`
`358`	`358`	`assert(success, "transferring unroll memory to storage memory failed")`
`359`	`359`	`}`
`360`	`360`	`entries.synchronized {`