Very WIP: Stage2 revival (#78)

* Very WIP: Stage2 revival

Needs some corresponding changes in Cthulhu

WIP

WIP

WIP

WIP

fix inferred type and return handling of rrule calls (#83)

add custom Cthulhu toggle to show ad-transformed code (#85)

add inlining policy for ADInterpreter (#84)

fixup

* Fix issues in untyped transform

Author: Keno

Project.toml

@@ -7,7 +7,9 @@ version = "0.1.1"
 ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
+Cthulhu = "f68482b8-f384-11e8-15f7-abe071a5a75f"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
+OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"

scratch/stage2.jl

@@ -0,0 +1,22 @@
+using Cthulhu
+using Diffractor
+using Diffractor: ADInterpreter
+using Diffractor: var"'", ∂⃖
+
+function foo(x)
+    sin(x)
+end
+bar(x) = foo(x)
+
+diffsin(x) = bar'(x)
+
+diffsin(1.0)
+
+function do_the_thing()
+    interp = ADInterpreter()
+    mi = Cthulhu.get_specialization(Tuple{map(Core.Typeof, (diffsin, 2.0))...})
+    Cthulhu.do_typeinf!(interp, mi)
+    (interp, mi)
+end
+(interp, mi) = do_the_thing();
+Diffractor.codegen(interp, Diffractor.ADCursor(0, mi))(1.0)

src/Diffractor.jl

@@ -17,6 +17,13 @@ include("stage1/recurse_fwd.jl")
 include("stage1/mixed.jl")
 include("stage1/broadcast.jl")
 
+include("stage2/interpreter.jl")
+include("stage2/lattice.jl")
+include("stage2/abstractinterpret.jl")
+include("stage2/tfuncs.jl")
+
+include("codegen/reverse.jl")
+
 include("extra_rules.jl")
 
 include("higher_fwd_rules.jl")