Increment stack counter inCompilationRoot only

hpi-swa · Feb 23, 2025 · 3da7a78 · 3da7a78
1 parent 51cc251
commit 3da7a78
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/de.hpi.swa.trufflesqueak/src/de/hpi/swa/trufflesqueak/nodes/StartContextRootNode.java b/src/de.hpi.swa.trufflesqueak/src/de/hpi/swa/trufflesqueak/nodes/StartContextRootNode.java
@@ -52,20 +52,22 @@ public StartContextRootNode(final SqueakLanguage language, final CompiledCodeObj
 
     @Override
     public Object execute(final VirtualFrame frame) {
+        final boolean inCompilationRoot = CompilerDirectives.inCompilationRoot();
         initializeFrame(frame);
         try {
-            if (++ApproximateStackDepth > MaximumStackDepth) {
+            if (inCompilationRoot && ++ApproximateStackDepth > MaximumStackDepth) {
                 throw ProcessSwitch.create(getGetOrCreateContextNode().executeGet(frame));
             }
-
             interruptHandlerNode.execute(frame);
             return executeBytecodeNode.execute(frame, initialPC);
         } catch (final NonVirtualReturn | ProcessSwitch nvr) {
             /* {@link getGetOrCreateContextNode()} acts as {@link BranchProfile} */
             getGetOrCreateContextNode().executeGet(frame).markEscaped();
             throw nvr;
         } finally {
-            --ApproximateStackDepth;
+            if (inCompilationRoot) {
+                --ApproximateStackDepth;
+            }
             materializeContextOnMethodExitNode.execute(frame);
         }
     }