comment

pytorch · Jan 5, 2021 · 9a03677 · 9a03677
1 parent 180e173
commit 9a03677
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/aten/src/ATen/native/cuda/Dropout.cu b/aten/src/ATen/native/cuda/Dropout.cu
@@ -78,6 +78,7 @@ fused_dropout_kernel_vec(at::cuda::detail::TensorInfo<scalar_t, IndexType> a,
     if ((VEC == 4) || (gridxvec_loop_state == 0)) {
       rand = curand_uniform4(&state);
     } else {
+      // sets up the last two values we generated last iteration to be used this iteration.
       rand.x = rand.z;
       rand.y = rand.w;
       gridxvec_loop_state ^= 1;