wip

Author: aviatesk

src/stage2/interpreter.jl

@@ -95,24 +95,22 @@ lower_level(interp::ADInterpreter) = change_level(interp, interp.current_level -
 disable_forward(interp::ADInterpreter) = ADInterpreter(interp; forward=false)
 
 function CC.InferenceState(result::InferenceResult, cache::Symbol, interp::ADInterpreter)
+    ret = @invoke CC.InferenceState(result::InferenceResult, cache::Symbol, interp::AbstractInterpreter)
+    ret === nothing && return ret
     if interp.current_level === missing
-        error()
+        # override initial bestguess
+        ret.bestguess = CC.typeinf_type(interp.native_interpreter, result.linfo, #=allow_extended=#true)
     end
-    return @invoke CC.InferenceState(result::InferenceResult, cache::Symbol, interp::AbstractInterpreter)
-    # prepare an InferenceState object for inferring lambda
-    world = get_world_counter(interp)
-    src = retrieve_code_info(result.linfo, world)
-    src === nothing && return nothing
-    validate_code_in_debug_mode(result.linfo, src, "lowered")
-    return InferenceState(result, src, cache, interp, Bottom)
+    return ret
 end
 
-
-function CC.initial_bestguess(interp::ADInterpreter, result::InferenceResult)
+function CC.update_bestguess!(interp::ADInterpreter, frame::InferenceState,
+                              currstate::CC.VarTable, @nospecialize(rt))
     if interp.current_level === missing
-        return CC.typeinf_lattice(interp.native_interpreter, result.linfo)
+        # TODO
     end
-    return Bottom
+    return @invoke CC.update_bestguess!(interp::AbstractInterpreter, frame::InferenceState,
+                                        currstate::CC.VarTable, rt::Any)
 end
 
 function Cthulhu.get_optimized_codeinst(interp::ADInterpreter, curs::ADCursor)
@@ -121,7 +119,6 @@ function Cthulhu.get_optimized_codeinst(interp::ADInterpreter, curs::ADCursor)
 end
 Cthulhu.AbstractCursor(interp::ADInterpreter, mi::MethodInstance) = ADCursor(0, mi, false)
 
-
 # This is a lie, but let's clean this up later
 Cthulhu.can_descend(interp::ADInterpreter, @nospecialize(key), optimize::Bool) = true