I am on Enzyme main (30b6b2d93d8ef1bdfb9f628e8c111d123cc4595e), CUDA master (7ff012) and Julia 1.10.6.
using Enzyme, CUDA
function f(a, b)
c = a .+ b
Array(c)[1]
end
a = CuArray(rand(5))
b = CuArray(rand(5))
da = zero(a)
db = zero(b)
f(a, b) # Works
autodiff(set_runtime_activity(Reverse), f, Active, Duplicated(a, da), Duplicated(b, db))
I am on Enzyme main (30b6b2d93d8ef1bdfb9f628e8c111d123cc4595e), CUDA master (7ff012) and Julia 1.10.6.