Zygote.jl
Zygote.jl copied to clipboard
CUDA complex valued broadcasting broken by dual_function not generating complex duals
example where I ran into this
ex = gpu(im*zeros(10))
y, back = pullback(x->abs.(x), ex)
back(y)
if I'm reading correctly, it seems to come down to ForwardDiff
not currently supporting complex numbers. Is there some way around this?