move check for previous compilation to the correct location (#58)

JuliaDiff · Mar 9, 2017 · 09e7471 · 09e7471
1 parent 1baad37
commit 09e7471
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/api/tape.jl b/src/api/tape.jl
@@ -111,14 +111,17 @@ stack has bubbled up to top level).
 """
 function compile(t::AbstractTape)
     ct = CompiledTape{gensym()}(t)
-    eval(current_module(), :(ReverseDiff.iscompiled($ct))) || compile(ct)
+    compile(ct)
     return ct
 end
 
 function compile(ct::CompiledTape)
-    eval(current_module(), generate_forward_pass_method(typeof(ct), ct.tape.tape))
-    eval(current_module(), generate_reverse_pass_method(typeof(ct), ct.tape.tape))
-    eval(current_module(), :(ReverseDiff.iscompiled(ct::$(typeof(ct))) = true))
+    previously_compiled = eval(current_module(), :(ReverseDiff.iscompiled($ct)))
+    if !(previously_compiled)
+        eval(current_module(), generate_forward_pass_method(typeof(ct), ct.tape.tape))
+        eval(current_module(), generate_reverse_pass_method(typeof(ct), ct.tape.tape))
+        eval(current_module(), :(ReverseDiff.iscompiled(ct::$(typeof(ct))) = true))
+    end
     return ct
 end