minor improvements to trace execution

NethermindEth · Sep 1, 2023 · ce855a8 · ce855a8
1 parent 76a574f
commit ce855a8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/vm/vm.go b/pkg/vm/vm.go
@@ -159,9 +159,10 @@ func (vm *VirtualMachine) Proof() ([]Context, []*f.Element, error) {
 
 	// update pc, fp and ap to be their real value
 	// that is, pc_i + 0, ap + len(programBytecode), fp + len(programBytecode)
+	totalBytecode := vm.MemoryManager.Memory.Segments[programSegment].Len()
 	for i := range vm.Trace {
-		vm.Trace[i].Ap += vm.MemoryManager.Memory.Segments[programSegment].Len()
-		vm.Trace[i].Fp += vm.MemoryManager.Memory.Segments[programSegment].Len()
+		vm.Trace[i].Ap += totalBytecode
+		vm.Trace[i].Fp += totalBytecode
 	}
 
 	// after that, get the relocated memory