Handle write-race conditions in CUDA kernels: Add atomic operation

Handle write-race conditions in CUDA kernels: Add atomic operation #4554