How to use `@atomic` with CUDA?

j = Float32(0.0) this fixes it.