cache AllocationReporter in PythonContext

Author: lukasstadler

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/PythonContext.java

@@ -107,6 +107,7 @@
 import com.oracle.truffle.api.TruffleLanguage;
 import com.oracle.truffle.api.TruffleLanguage.Env;
 import com.oracle.truffle.api.TruffleLogger;
+import com.oracle.truffle.api.instrumentation.AllocationReporter;
 import com.oracle.truffle.api.interop.ExceptionType;
 import com.oracle.truffle.api.interop.InteropLibrary;
 import com.oracle.truffle.api.interop.UnsupportedMessageException;
@@ -314,6 +315,7 @@ private static final class AtExitHook {
     private final ThreadLocal<ArrayDeque<String>> currentImport = new ThreadLocal<>();
 
     @CompilationFinal(dimensions = 1) private Object[] optionValues;
+    private AllocationReporter allocationReporter;
 
     /*
      * These maps are used to ensure that each "deserialization" of code in the parser gets a
@@ -341,6 +343,13 @@ public PythonContext(PythonLanguage language, TruffleLanguage.Env env, Python3Co
         this.err = env.err();
     }
 
+    public AllocationReporter getAllocationReporter() {
+        if (allocationReporter == null) {
+            return allocationReporter = env.lookup(AllocationReporter.class);
+        }
+        return allocationReporter;
+    }
+
     public ThreadGroup getThreadGroup() {
         return threadGroup;
     }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/object/PythonObjectFactory.java

@@ -220,7 +220,7 @@ static PythonLanguage getLanguage(boolean marker, double marker2,
     }
 
     protected static AllocationReporter getAllocationReporter(ContextReference<PythonContext> contextRef) {
-        return contextRef.get().getEnv().lookup(AllocationReporter.class);
+        return contextRef.get().getAllocationReporter();
     }
 
     public final PythonLanguage getLanguage() {