special case AbstractWirtinger in at_thunk

Author: simeonschaub

src/differentials.jl

@@ -265,6 +265,12 @@ struct Thunk{F} <: AbstractThunk
 end
 
 macro thunk(body)
+    if body isa Expr && body.head == :call
+        fname = body.args[1]
+        if fname in (:Wirtinger, :ComplexGradient)
+            return :($fname($((:(@thunk $i) for i in body.args[2:end])...)))
+        end
+    end
     return :(Thunk(() -> $(esc(body))))
 end
 

test/differentials.jl

@@ -89,7 +89,7 @@
         @test refine_differential(typeof([1.2]), Wirtinger(2,2)) == 4
 
         # For most differentials, in most domains, this does nothing
-        for der in (DNE(), @thunk(23), @thunk(Wirtinger(2,2)), [1 2], One(), Zero(), 0.0)
+        for der in (DNE(), @thunk(23), [1 2], One(), Zero(), 0.0)
             for 𝒟 in typeof.((1.0 + 1im, [1.0 + 1im], 1.2, [1.2]))
                 @test refine_differential(𝒟, der) === der
             end