fix[venom]: fix _stack_reorder() routine

vyper/venom/venom_to_assembly.py

@@ -1,4 +1,5 @@
-from collections import Counter
+from bisect import insort
+from collections import Counter, defaultdict
 from typing import Any
 
 from vyper.exceptions import CompilerPanic, StackTooDeep
@@ -210,11 +211,19 @@ def _stack_reorder(
 
         counts = Counter(stack_ops)
 
+        positions: dict[IROperand, list[int]] = defaultdict(lambda: [])
+        for op in stack_ops:
+            positions[op] = []
+            for i in range(counts[op]):
+                positions[op].append(stack.get_depth(op, i + 1))
+
         for i in range(stack_ops_count):
             op = stack_ops[i]
             final_stack_depth = -(stack_ops_count - i - 1)
-            depth = stack.get_depth(op, counts[op])  # type: ignore
-            counts[op] -= 1
+            depth = positions[op].pop()  # type: ignore
+            assert depth not in range(
+                -stack_ops_count + 1, final_stack_depth
+            ), f"{depth} : ({-stack_ops_count - 1}, {final_stack_depth})"
 
             if depth == StackModel.NOT_IN_STACK:
                 raise CompilerPanic(f"Variable {op} not in stack")
@@ -225,9 +234,23 @@ def _stack_reorder(
             if op == stack.peek(final_stack_depth):
                 continue
 
+            if len(positions[stack.peek(0)]) != 0:
+                positions[stack.peek(0)].remove(0)
+                insort(positions[stack.peek(0)], depth)
+
             cost += self.swap(assembly, stack, depth)
+            if final_stack_depth in positions[stack.peek(final_stack_depth)]:
+                positions[stack.peek(final_stack_depth)].remove(final_stack_depth)
+                positions[stack.peek(final_stack_depth)].insert(0, 0)
+            else:
+                positions[stack.peek(final_stack_depth)].insert(0, 0)
+
             cost += self.swap(assembly, stack, final_stack_depth)
 
+        assert (
+            len(stack_ops) == 0 or stack._stack[-len(stack_ops) :] == stack_ops
+        ), f"Wrong stack {stack} {stack_ops}"
+
         return cost
 
     def _emit_input_operands(