From 9a03677c6f56ed27b57e9a79c938adb8f6065a8d Mon Sep 17 00:00:00 2001
From: Michael Carilli <mcarilli@gmail.com>
Date: Tue, 5 Jan 2021 15:10:34 -0700
Subject: [PATCH] comment

---
 aten/src/ATen/native/cuda/Dropout.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aten/src/ATen/native/cuda/Dropout.cu b/aten/src/ATen/native/cuda/Dropout.cu
index 27bbdb3f115d..c3e456d97056 100644
--- 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;