Skip to content
Merged
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions rust/lance-datafusion/src/exec.rs
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,7 @@ impl std::fmt::Debug for LanceExecutionOptions {
}
}

const DEFAULT_LANCE_MEM_POOL_SIZE_PER_PARTITION: u64 = 100 * 1024 * 1024;
const DEFAULT_LANCE_MEM_POOL_SIZE_PER_PARTITION: u64 = 150 * 1024 * 1024;
const DEFAULT_LANCE_MAX_TEMP_DIRECTORY_SIZE: u64 = 100 * 1024 * 1024 * 1024; // 100GB

impl LanceExecutionOptions {
Expand Down Expand Up @@ -360,7 +360,9 @@ impl LanceExecutionOptions {
}

pub fn new_session_context(options: &LanceExecutionOptions) -> SessionContext {
let mut session_config = SessionConfig::new();
let sort_spill_reservation_bytes = (options.mem_pool_size() / 3) as usize;
let mut session_config =
SessionConfig::new().with_sort_spill_reservation_bytes(sort_spill_reservation_bytes);
let mut runtime_env_builder = RuntimeEnvBuilder::new();
if let Some(target_partition) = options.target_partition {
session_config = session_config.with_target_partitions(target_partition);
Expand Down
Loading