Skip to content
This repository has been archived by the owner on Sep 18, 2023. It is now read-only.

Commit

Permalink
fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
zhztheplayer committed Sep 24, 2021
1 parent 9875fd0 commit d883575
Showing 1 changed file with 2 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@ import com.intel.oap.spark.sql.execution.datasources.v2.arrow._
import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream
import org.apache.arrow.dataset.jni.NativeMemoryPool
import org.apache.arrow.memory.AllocationListener
import org.apache.arrow.memory.BaseAllocator.configBuilder
import org.apache.arrow.memory.BufferAllocator
import org.apache.arrow.memory.ImmutableConfig
import org.apache.arrow.memory.MemoryChunkCleaner
Expand Down Expand Up @@ -88,7 +87,7 @@ object SparkMemoryUtils extends Logging {

val defaultAllocator: BufferAllocator = {
val alloc = new RootAllocator(
configBuilder()
RootAllocator.configBuilder()
.maxAllocation(Long.MaxValue)
.memoryChunkManagerFactory(memoryChunkManagerFactory)
.listener(allocListener)
Expand Down Expand Up @@ -238,7 +237,7 @@ object SparkMemoryUtils extends Logging {
}

private val allocator = new RootAllocator(
configBuilder()
RootAllocator.configBuilder()
.maxAllocation(SQLConf.get.getConf(MEMORY_OFFHEAP_SIZE))
.memoryChunkManagerFactory(MemoryChunkCleaner.newFactory())
.listener(MemoryChunkCleaner.gcTrigger())
Expand Down

0 comments on commit d883575

Please sign in to comment.