Enabling custom block size schedule.

This commit is contained in:
Nicolas Patry 2024-05-30 05:17:00 +00:00
parent cf59593454
commit 13caf958eb

View File

@ -77,6 +77,9 @@ impl Infer {
false
};
let block_size = if flashdecoding { 256 } else { 32 };
let block_size = std::env::var("BLOCK_SIZE")
.map(|b| b.parse().unwrap_or(block_size))
.unwrap_or(block_size);
let queue = Queue::new(requires_padding, block_size, window_size, speculate);
let shared = Arc::new(Shared {
batching_task: Notify::new(),