feat: update segmentation weights for hardware memory

[jonathanpwang]

segment_ctx.rs:

• DEFAULT_MAX_CELLS → DEFAULT_MAX_MEMORY = 15gb
• max_cells → max_memory in SegmentationLimits
• set_max_cells → set_max_memory

ctx.rs:

• with_max_cells → with_max_memory

metered_cost.rs:

• Updated import to use DEFAULT_MAX_MEMORY

cli/src/commands/prove.rs:

• Updated import to DEFAULT_MAX_MEMORY
• segment_max_cells → segment_max_memory
• with_max_cells → with_max_memory

benchmarks/prove/src/bin/async_regex.rs:

• segment_max_cells → segment_max_memory
• set_max_cells → set_max_memory

benchmarks/prove/src/util.rs:

• segment_max_cells → segment_max_memory
• set_max_cells → set_max_memory

[shuklaayush]

looks good but would prefer if we first do a reth benchmark run to confirm that 1.2bn cells ~ 15gb

[Copilot]

Pull request overview

This PR refactors the segmentation memory model from cell-based counting to actual hardware memory-based calculations. The changes introduce weight multipliers for different cell types and convert the final calculation to bytes, providing more accurate memory tracking for segmentation decisions.

Key changes:

• Replaced max_cells (1.2B cells) with max_memory (15GB) as the segmentation threshold
• Introduced main_cell_weight and interaction_cell_weight multipliers to account for proof system overhead
• Updated memory calculation formula to: (main_cells * main_weight + interaction_cells * interaction_weight) * base_field_size

Reviewed changes

Copilot reviewed 5 out of 6 changed files in this pull request and generated 1 comment.

Show a summary per file

File	Description
crates/vm/src/arch/execution_mode/metered/segment_ctx.rs	Core refactoring: renamed constants and fields, added weight multipliers and base field size, updated memory calculation formula with weighted cell counts
crates/vm/src/arch/execution_mode/metered/ctx.rs	Renamed builder method with_max_cells to with_max_memory, added builder methods for new weight configuration fields
crates/vm/src/arch/execution_mode/metered_cost.rs	Updated import from DEFAULT_MAX_CELLS to DEFAULT_MAX_MEMORY and corresponding usage in cost calculation
crates/cli/src/commands/prove.rs	Updated CLI argument from segment_max_cells to segment_max_memory, improved help text to clarify memory measurement in bytes
benchmarks/prove/src/util.rs	Updated CLI argument and setter method names to use memory terminology instead of cells
benchmarks/prove/src/bin/async_regex.rs	Updated CLI argument and setter method names to use memory terminology instead of cells

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

[jonathanpwang]

looks good but would prefer if we first do a reth benchmark run to confirm that 1.2bn cells ~ 15gb

I ran it and it did seem to give approximately similar behavior.