Runtime prints outbound function dependencies in verbose mode.

Author: braxtonmckee

typed_python/compiler/python_to_native_converter.py

@@ -925,6 +925,11 @@ def _installInflightFunctions(self, name):
 
                     nativeFunction, actual_output_type = self._inflight_definitions.get(identifier)
 
+                    outboundTargets = []
+                    for outboundFuncId in self._dependencies.getNamesDependedOn(identifier):
+                        name = self._link_name_for_identity[outboundFuncId]
+                        outboundTargets.append(self._targets[name])
+
                     try:
                         v.onNewFunction(
                             identifier,
@@ -938,7 +943,8 @@ def _installInflightFunctions(self, name):
                             functionConverter._varname_to_type.get(FunctionOutput),
                             functionConverter._varname_to_type.get(FunctionYield),
                             {k: v.typeRepresentation for k, v in functionConverter._varname_to_type.items() if isinstance(k, str)},
-                            conversionType
+                            conversionType,
+                            outboundTargets
                         )
                     except Exception:
                         logging.exception("event handler %s threw an unexpected exception", v.onNewFunction)

typed_python/compiler/runtime.py

@@ -56,7 +56,8 @@ def onNewFunction(
         outputType,
         yieldType,
         variableTypes,
-        conversionType
+        conversionType,
+        calledFunctions,
     ):
         pass
 
@@ -96,7 +97,8 @@ def onNewFunction(
         outputType,
         yieldType,
         variableTypes,
-        conversionType
+        conversionType,
+        calledFunctions,
     ):
         if self.short:
             print(
@@ -127,6 +129,11 @@ def onNewFunction(
                 else:
                     print("        ", varname, varVal)
 
+            if calledFunctions:
+                print("    calls:")
+                for callTarget in calledFunctions:
+                    print("        ", callTarget)
+
 
 class CountCompilationsVisitor(RuntimeEventVisitor):
     def __init__(self):
@@ -145,7 +152,8 @@ def onNewFunction(
         outputType,
         yieldType,
         variableTypes,
-        conversionType
+        conversionType,
+        calledFunctions,
     ):
         self.count += 1
 

typed_python/compiler/tests/closures_test.py

@@ -52,7 +52,8 @@ def onNewFunction(
         outputType,
         yieldType,
         variableTypes,
-        conversionType
+        conversionType,
+        calledFunctions,
     ):
         self.didCompile = True
 

typed_python/compiler/tests/conversion_test.py

@@ -89,7 +89,8 @@ def onNewFunction(
         outputType,
         yieldType,
         variableTypes,
-        conversionType
+        conversionType,
+        calledFunctions,
     ):
         self.types[funcName] = makeNamedTuple(
             inputTypes=inputTypes,
@@ -1405,7 +1406,8 @@ def onNewFunction(
                 outputType,
                 yieldType,
                 variableTypes,
-                conversionType
+                conversionType,
+                calledFunctions,
             ):
                 assert issubclass(outputType.typeRepresentation, Type)
 

typed_python/compiler/tests/entrypoint_test.py

@@ -422,20 +422,27 @@ def onNewFunction(
                 outputType,
                 yieldType,
                 variableTypes,
-                conversionType
+                conversionType,
+                calledFunctions,
             ):
-                out[funcName] = (inputTypes, outputType, variableTypes)
+                out[funcName] = (inputTypes, outputType, variableTypes, calledFunctions)
+
+        @Entrypoint
+        def g(x):
+            return x
 
         @Entrypoint
         def f(x):
-            y = x + 1
+            y = x + g(1)
             return str(y)
 
         with Visitor():
             f.resultTypeFor(int)
 
         self.assertTrue('f' in out, out)
         self.assertEqual(out['f'][2]['y'], int)
+        self.assertTrue(out['f'][3])
+        assert type(out['f'][3][0]).__name__ == 'TypedCallTarget'
 
     def test_star_args_on_entrypoint(self):
         @Entrypoint

typed_python/compiler/type_wrappers/alternative_wrapper.py

@@ -355,6 +355,7 @@ def convert_attribute(self, context, instance, attribute, nocheck=False):
                 return self.convert_method_call(context, instance, "__getattr__", (context.constant(attribute),), {})
 
             return super().convert_attribute(context, instance, attribute)
+
         if len(validIndices) == 1:
             with context.ifelse(instance.nonref_expr.ElementPtrIntegers(0, 1).load().neq(validIndices[0])) as (then, otherwise):
                 with then:

typed_python/type_function_test.py

@@ -360,7 +360,8 @@ def onNewFunction(
                 outputType,
                 yieldType,
                 variableTypes,
-                conversionType
+                conversionType,
+                calledFunctions,
             ):
                 if funcName == "do":
                     self.variableTypes = variableTypes