revert f978017

Author: MasonProtter

src/infer.jl

@@ -86,7 +86,6 @@ mutable struct Frame
   edges::Vector{Any}
   stmts::Vector{Vector{Variable}}
   rettype::AType
-  outer::Bool
 end
 
 getblock(fr::Frame, b, follow) =
@@ -96,7 +95,7 @@ getblock(fr::Frame, b, follow) =
 # TODO clear up inlined edges
 uninline!(fr::Frame, b, follow) = deleteat!(fr.inlined[b], follow+1:length(fr.inlined[b]))
 
-Frame(ir::IR; outer = false) = Frame(ir, [IR[] for _ = 1:length(blocks(ir))], [], keys.(blocks(ir)), Union{}, outer)
+Frame(ir::IR) = Frame(ir, [IR[] for _ = 1:length(blocks(ir))], [], keys.(blocks(ir)), Union{})
 
 function frame(ir::IR, args...)
   prepare_ir!(ir)
@@ -201,7 +200,7 @@ function step!(inf::Inference)
       error("Unrecognised expression $(st.expr)")
     end
   elseif (brs = openbranches(block); length(brs) == 1 && !isreturn(brs[1])
-          && !(frame.outer && brs[1].block == length(frame.ir.blocks)))
+          && !(brs[1].block == length(frame.ir.blocks)))
     inferbranch!(inf, frame, b, f, brs[1])
   else
     for br in brs
@@ -250,19 +249,3 @@ function return_type(ir::IR, args...)
   infer!(inf)
   return fr.rettype
 end
-
-function inlineable(ir::IR, inf::Inference)
-  for (v, st) in ir
-    Ts = exprtype.((ir,), st.expr.args)
-    haskey(inf.frames, Ts) && return (v, Ts)
-  end
-end
-
-function inlineall(ir::IR, inf::Inference)
-  while (next = inlineable(ir, inf)) != nothing
-    v, Ts = next
-    subir = inf.frames[Ts].ir
-    ir = inline(ir, v, subir)
-  end
-  return ir
-end

src/trace.jl

@@ -133,11 +133,7 @@ function abstract!(tr, bl, env)
   for i = 1:length(args)
     argtypes(ir)[i] = exprtype(tr.ir, rename(env, args[i]))
   end
-  inf = Inference(Frame(ir, outer = true), tr.primitives)
-  infer!(inf)
-  # TODO weird that we need to expand here; perhaps inlining is impicitly pruning
-  # somehow.
-  ir = inlineall(ir, inf) |> expand!
+  infer!(Inference(Frame(ir), tr.primitives))
   inline!(tr.ir, ir, rename.((env,), args))
   for (k, v) in zip(arguments(block(bl.ir, after)), arguments(blocks(tr.ir)[end]))
     env[k] = v