closer?

Author: mcabbott

src/stage1/forward.jl

@@ -118,7 +118,12 @@ function (::∂☆internal{1})(args::AbstractTangentBundle{1}...)
     end
 end
 
-ChainRulesCore.frule_via_ad(::DiffractorRuleConfig, args...) = ∂☆internal{1}()(args...)
+function ChainRulesCore.frule_via_ad(::DiffractorRuleConfig, partials, args...)
+    tangents = map(partials, args) do p, a
+        TangentBundle{1}(a, (p,))
+    end
+    ∂☆internal{1}()(tangents...)
+end
 
 function (::∂☆shuffle{N})(args::AbstractTangentBundle{N}...) where {N}
     ∂☆p = ∂☆{minus1(N)}()

test/runtests.jl

@@ -206,8 +206,8 @@ y45, back45 = rrule_via_ad(DiffractorRuleConfig(), x -> log(exp(x)), 2.0)
 @test y45 ≈ 2.0
 @test back45(1) == (ZeroTangent(), 1.0)
 
-# z45, delta45 = frule_via_ad(DiffractorRuleConfig(), (0,1), x -> log(exp(x)), 2.0)
-# @test z45 ≈ 2.0
-# @test delta45 ≈ 1.0
+z45, delta45 = frule_via_ad(DiffractorRuleConfig(), (0,1), x -> log(exp(x)), 2.0)
+@test z45 ≈ 2.0
+@test delta45 ≈ 1.0
 
 include("pinn.jl")