Skip to content

v0.47.0: Memory-budget controls for predict and predict_incremental#67

Merged
sergeyf merged 9 commits intomainfrom
v0.47.0
Mar 26, 2026
Merged

v0.47.0: Memory-budget controls for predict and predict_incremental#67
sergeyf merged 9 commits intomainfrom
v0.47.0

Conversation

@atalyaalon
Copy link
Copy Markdown
Contributor

@atalyaalon atalyaalon commented Mar 26, 2026

Summary

  • Add configurable max_chunk_pairs to predict() and predict_incremental() to cap Phase A chunk size (default 1M, 0 = unlimited)
  • Propagate total_ram_bytes through phase-split and subblocked predict paths
  • Disable RUST_BATCH_BASE_CHUNK_PAIRS hardcoded floor; rely on memory-budget-derived limits
  • Simplify chunk_pairs calculation in compute_rust_batch_chunk_plan
  • Revise threading docs: deployment env patterns, BLAS/NumExpr pinning guidance
  • Update README and subclustering docs with max_chunk_pairs usage

Test plan

  • Existing test_cluster_incremental.py updated to match new predict_helper signature

@atalyaalon atalyaalon force-pushed the v0.47.0 branch 9 times, most recently from b46cfd7 to d4d8759 Compare March 26, 2026 20:54
@atalyaalon atalyaalon requested a review from sergeyf March 26, 2026 20:58
…sizing

- Add configurable max_chunk_pairs to predict() and predict_incremental()
  to cap Phase A chunk size (default 1M, 0 = unlimited)
- Propagate total_ram_bytes through phase-split and subblocked predict paths
- Disable RUST_BATCH_BASE_CHUNK_PAIRS hardcoded floor; rely on memory-budget
- Simplify chunk_pairs calculation in compute_rust_batch_chunk_plan
- Update docs: threading guidance, max_chunk_pairs usage, subclustering

Bump to v0.47.0
@sergeyf sergeyf merged commit 0b0971e into main Mar 26, 2026
18 checks passed
@sergeyf sergeyf deleted the v0.47.0 branch March 26, 2026 21:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants