fix correlation1D illegal cuda memory bug

TuSimple · Dec 11, 2018 · f0213d0 · f0213d0
1 parent 1c561f7
commit f0213d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/operator/correlation1D.cu b/src/operator/correlation1D.cu
@@ -290,7 +290,7 @@ void Forward_gpu(
 
     int x_shift = - neighborhood_grid_radius_;
     if(single_side == -1) { // to the left
-      x_shift = -neighborhood_grid_width_;
+      x_shift = -neighborhood_grid_radius_;
     } else if(single_side == 1) { // to the right
       x_shift = 0;
     }
@@ -338,7 +338,7 @@ void Backward_gpu(
 
     int x_shift = - neighborhood_grid_radius_;
     if (single_side == -1) { // to the left
-      x_shift = -neighborhood_grid_width_;
+      x_shift = -neighborhood_grid_radius_;
     } else if(single_side == 1) { // to the right
       x_shift = 0;
     }