Update segment used offsets with output size. (#1681)

Co-authored-by: Pedro Fontana <[email protected]>
lambdaclass · Mar 20, 2024 · 0b6d116 · 0b6d116
1 parent 16a844e
commit 0b6d116
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/vm/src/vm/vm_core.rs b/vm/src/vm/vm_core.rs
@@ -1107,7 +1107,10 @@ impl VirtualMachine {
     #[doc(hidden)]
     pub fn set_output_stop_ptr_offset(&mut self, offset: usize) {
         if let Some(BuiltinRunner::Output(builtin)) = self.builtin_runners.first_mut() {
-            builtin.set_stop_ptr_offset(offset)
+            builtin.set_stop_ptr_offset(offset);
+            if let Some(segment_used_sizes) = &mut self.segments.segment_used_sizes {
+                segment_used_sizes[builtin.base()] = offset;
+            }
         }
     }
 }