Francois' fix

Consensys · Oct 11, 2024 · 1d183f8 · 1d183f8
1 parent d6535db
commit 1d183f8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/...metization/src/main/java/net/consensys/linea/zktracer/module/hub/defer/DeferRegistry.java b/...metization/src/main/java/net/consensys/linea/zktracer/module/hub/defer/DeferRegistry.java
@@ -220,4 +220,8 @@ public void resolveUponContextReEntry(Hub hub, CallFrame callFrame) {
       contextReEntryDefers.remove(callFrame);
     }
   }
+
+  public void unscheduleForContextReEntry(ContextReEntryDefer defer, CallFrame callFrame) {
+    contextReEntryDefers.get(callFrame).remove(defer);
+  }
 }
diff --git a/...nsensys/linea/zktracer/module/hub/section/call/precompileSubsection/ModexpSubsection.java b/...nsensys/linea/zktracer/module/hub/section/call/precompileSubsection/ModexpSubsection.java
@@ -73,6 +73,7 @@ public ModexpSubsection(final Hub hub, final CallSection callSection) {
                 .compareTo(BigInteger.valueOf(PROVER_MAX_INPUT_BYTE_SIZE))
             >= 0) {
       hub.modexpEffectiveCall().addPrecompileLimit(Integer.MAX_VALUE);
+      hub.defers().unscheduleForContextReEntry(this, hub.currentFrame());
       return;
     }