use a singleContextAssumption to cache class objects in GetClassNode

Author: timfel

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/PythonLanguage.java

@@ -53,6 +53,7 @@
 import com.oracle.graal.python.runtime.PythonOptions;
 import com.oracle.graal.python.runtime.PythonParser.ParserMode;
 import com.oracle.graal.python.runtime.exception.PException;
+import com.oracle.truffle.api.Assumption;
 import com.oracle.truffle.api.CallTarget;
 import com.oracle.truffle.api.CompilerAsserts;
 import com.oracle.truffle.api.CompilerDirectives;
@@ -89,6 +90,8 @@ public final class PythonLanguage extends TruffleLanguage<PythonContext> {
     public static final String MIME_TYPE = "text/x-python";
     public static final String EXTENSION = ".py";
 
+    public static Assumption singleContextAssumption = Truffle.getRuntime().createAssumption("Only a single context is active");
+
     @CompilationFinal private boolean nativeBuildTime = TruffleOptions.AOT;
     private final NodeFactory nodeFactory;
 
@@ -394,4 +397,10 @@ private static Source newSource(PythonContext ctxt, SourceBuilder srcBuilder, St
     public boolean isNativeBuildTime() {
         return nativeBuildTime;
     }
+
+    @Override
+    protected void initializeMultipleContexts() {
+        super.initializeMultipleContexts();
+        singleContextAssumption.invalidate();
+    }
 }

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

@@ -49,6 +49,7 @@
 import com.oracle.graal.python.runtime.exception.PException;
 import com.oracle.graal.python.runtime.exception.PythonErrorType;
 import com.oracle.graal.python.runtime.object.PythonObjectFactory;
+import com.oracle.truffle.api.Assumption;
 import com.oracle.truffle.api.CompilerDirectives;
 import com.oracle.truffle.api.CompilerDirectives.CompilationFinal;
 import com.oracle.truffle.api.TruffleLanguage.ContextReference;
@@ -101,4 +102,8 @@ public final PythonContext getContext() {
         }
         return contextRef.get();
     }
+
+    protected Assumption singleContextAssumption() {
+        return PythonLanguage.singleContextAssumption;
+    }
 }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/object/GetClassNode.java

@@ -74,6 +74,12 @@ public static GetClassNode create() {
 
     public abstract PythonClass execute(Object object);
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") GetSetDescriptor object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @Specialization
     protected PythonClass getIt(@SuppressWarnings("unused") GetSetDescriptor object) {
         return getCore().lookupType(PythonBuiltinClassType.GetSetDescriptor);
@@ -85,47 +91,95 @@ protected PythonClass getIt(PythonObject object,
         return profile.profile(object.getPythonClass());
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") PNone object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @Specialization
     protected PythonClass getIt(@SuppressWarnings("unused") PNone object) {
         return getCore().lookupType(PythonBuiltinClassType.PNone);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") PNotImplemented object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(PNotImplemented object) {
         return getCore().lookupType(PythonBuiltinClassType.PNotImplemented);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") PEllipsis object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(PEllipsis object) {
         return getCore().lookupType(PythonBuiltinClassType.PEllipsis);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") boolean object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(boolean object) {
         return getCore().lookupType(PythonBuiltinClassType.Boolean);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") int object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(int object) {
         return getCore().lookupType(PythonBuiltinClassType.PInt);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") long object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(long object) {
         return getCore().lookupType(PythonBuiltinClassType.PInt);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") double object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(double object) {
         return getCore().lookupType(PythonBuiltinClassType.PFloat);
     }
 
+    @Specialization(assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") String object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization
     protected PythonClass getIt(String object) {
@@ -143,6 +197,12 @@ protected PythonClass getIt(PythonNativeObject object,
         return getNativeClassNode.execute(object);
     }
 
+    @Specialization(guards = "isForeignObject(object)", assumptions = "singleContextAssumption()")
+    protected PythonClass getIt(@SuppressWarnings("unused") TruffleObject object,
+                    @Cached("getIt(object)") PythonClass klass) {
+        return klass;
+    }
+
     @SuppressWarnings("unused")
     @Specialization(guards = "isForeignObject(object)")
     protected PythonClass getIt(TruffleObject object) {