Fixes for the integration tests

NethermindEth · Feb 21, 2025 · ef6d244 · ef6d244
1 parent 795c82a
commit ef6d244
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pkg/runner/runner.go b/pkg/runner/runner.go
@@ -441,8 +441,10 @@ func (runner *Runner) RunFor(steps uint64) error {
 // Since this vm always finishes the run of the program at the number of steps that is a power of two in the proof mode,
 // there is no need to run additional steps before the loop.
 func (runner *Runner) EndRun() error {
-	if err := runner.RelocateTemporarySegments(); err != nil {
-		return err
+	if runner.runnerMode == ProofModeCairo {
+		if err := runner.RelocateTemporarySegments(); err != nil {
+			return err
+		}
 	}
 	for runner.checkUsedCells() != nil {
 		pow2Steps := utils.NextPowerOfTwo(runner.vm.Step + 1)

diff --git a/pkg/vm/memory/memory.go b/pkg/vm/memory/memory.go
@@ -469,7 +469,7 @@ func (memory *Memory) AddRelocationRule(segmentIndex int, addr MemoryAddress) {
 
 func (memory *Memory) RelocateTemporarySegments() error {
 	if len(memory.relocationRules) == 0 || len(memory.TemporarySegments) == 0 {
-		return fmt.Errorf("no relocation rules or temporary segments")
+		return nil
 	}
 	for i, segment := range memory.Segments {
 		for j := uint64(0); j < segment.RealLen(); j++ {