Zygote.jl icon indicating copy to clipboard operation
Zygote.jl copied to clipboard

CUDA complex valued broadcasting broken by dual_function not generating complex duals

Open dsweber2 opened this issue 4 years ago • 0 comments

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?

dsweber2 avatar Jul 13 '20 20:07 dsweber2