-
Notifications
You must be signed in to change notification settings - Fork 68
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
atomic operations error #421
Labels
upstream
Enzyme proper
Comments
I ran into this too, my MWE is: using Enzyme, CUDA
function mul_kernel(A)
i = threadIdx().x
if i <= length(A)
CUDA.@atomic A[i] *= A[i]
end
return nothing
end
function grad_mul_kernel(A, dA)
Enzyme.autodiff_deferred(mul_kernel, Const, Duplicated(A, dA))
return nothing
end
A = CUDA.ones(64,)
dA = similar(A)
dA .= 1
@cuda threads=length(A) grad_mul_kernel(A, dA)
dA It works fine without
|
Hello I understand that you may have other problems with higher priorities this is huge project that you are working on, but anybody had time to look into this atomic add problem ? |
Not yet unfortunately |
This has now landed on main. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
While trying to add atomically from kernel compilation error occur
System Ubuntu 20.0
Julia version 1.7.3
CUDA.jl :latest
Enzyme.jl : latest
GPU RTX 3080
kernel
enzyme function
invoking function
creating test data
error
The text was updated successfully, but these errors were encountered: