[GR-18397] Count method size at most once.

PullRequest: graalpython/666

Author: lukasstadler

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/PRootNode.java

@@ -48,6 +48,7 @@
 import com.oracle.truffle.api.TruffleLanguage;
 import com.oracle.truffle.api.frame.FrameDescriptor;
 import com.oracle.truffle.api.nodes.Node;
+import com.oracle.truffle.api.nodes.NodeUtil;
 import com.oracle.truffle.api.nodes.RootNode;
 import com.oracle.truffle.api.profiles.ConditionProfile;
 
@@ -64,6 +65,8 @@ public abstract class PRootNode extends RootNode {
      */
     @CompilationFinal private Assumption dontNeedExceptionState = createExceptionStateAssumption();
 
+    private int nodeCount = -1;
+
     protected PRootNode(TruffleLanguage<?> language) {
         super(language);
     }
@@ -72,6 +75,15 @@ protected PRootNode(TruffleLanguage<?> language, FrameDescriptor frameDescriptor
         super(language, frameDescriptor);
     }
 
+    public final int getNodeCount() {
+        CompilerAsserts.neverPartOfCompilation();
+        int n = nodeCount;
+        if (n != -1) {
+            return n;
+        }
+        return nodeCount = NodeUtil.countNodes(this);
+    }
+
     public ConditionProfile getFrameEscapedProfile() {
         return frameEscaped;
     }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/call/special/CallSpecialMethodNode.java

@@ -45,6 +45,7 @@
 import com.oracle.graal.python.builtins.objects.function.PFunction;
 import com.oracle.graal.python.builtins.objects.method.PBuiltinMethod;
 import com.oracle.graal.python.builtins.objects.method.PMethod;
+import com.oracle.graal.python.nodes.PRootNode;
 import com.oracle.graal.python.nodes.function.BuiltinFunctionRootNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
 import com.oracle.graal.python.nodes.function.builtins.PythonBinaryBuiltinNode;
@@ -92,7 +93,8 @@ private <T extends PythonBuiltinBaseNode> T getBuiltin(PBuiltinFunction func, Cl
     private <T extends PythonBuiltinBaseNode> boolean callerExceedsMaxSize(T builtinNode) {
         CompilerAsserts.neverPartOfCompilation();
         if (!maxSizeExceeded) {
-            int n = NodeUtil.countNodes(getRootNode());
+            RootNode root = getRootNode();
+            int n = root instanceof PRootNode ? ((PRootNode) root).getNodeCount() : NodeUtil.countNodes(root);
             // nb: option 'BuiltinsInliningMaxCallerSize' is defined as a compatible option, i.e.,
             // ASTs will only we shared between contexts that have the same value for this option.
             int maxSize = PythonOptions.getOption(lookupContextReference(PythonLanguage.class).get(), PythonOptions.BuiltinsInliningMaxCallerSize);