fix cuda bug

rusty1s · Jul 27, 2019 · 4fa1fb2 · 4fa1fb2
1 parent acd094e
commit 4fa1fb2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cuda/rw_kernel.cu b/cuda/rw_kernel.cu
@@ -18,9 +18,9 @@ __global__ void uniform_rw_kernel(
     out[n] = start[n];
 
     for (ptrdiff_t l = 1; l <= walk_length; l++) {
-      auto i = l * numel + n;
-      auto cur = out[(l - 1) * numel + n];
-      out[i] = col[row[cur] + int64_t(rand[i] * deg[cur])];
+      auto i = (l - 1) * numel + n;
+      auto cur = out[i];
+      out[l * numel + n] = col[row[cur] + int64_t(rand[i] * deg[cur])];
     }
   }
 }