splatting for dy

Author: AstitvaAggarwal

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/differentiate_with.jl

@@ -8,7 +8,7 @@ function Mooncake.rrule!!(dw::CoDual{<:DI.DifferentiateWith}, x::CoDual{<:Number
     y = f(primal_x)
 
     function pullback!!(dy)
-        tx = DI.pullback(f, backend, primal_x, (dy,))
+        tx = DI.pullback(f, backend, primal_x, (dy...,))
         return NoRData(), only(tx)
     end
 
@@ -23,7 +23,7 @@ function Mooncake.rrule!!(dw::CoDual{<:DI.DifferentiateWith}, x::CoDual{<:Abstra
     y = f(primal_x)
 
     function pullback!!(dy)
-        tx = DI.pullback(f, backend, primal_x, (dy,))
+        tx = DI.pullback(f, backend, primal_x, (dy...,))
         fdata_arg .+= only(tx)
         return NoRData(), NoRData()
     end