fix illegal memory access

rusty1s · rusty1s · commit 5f855a28e56d · 2020-03-26T17:46:42.000+01:00
diff --git a/csrc/cuda/fps_cuda.cu b/csrc/cuda/fps_cuda.cu
@@ -25,6 +25,7 @@ __global__ void fps_kernel(const scalar_t *src, const int64_t *ptr,
   }
 
   for (int64_t m = out_ptr[batch_idx] + 1; m < out_ptr[batch_idx + 1]; m++) {
+    __syncthreads();
     int64_t old = out[m - 1];
 
     scalar_t best = (scalar_t)-1.;

Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,7 @@ __global__ void fps_kernel(const scalar_t src, const int64_t ptr,`
`25`	`25`	`}`
`26`	`26`
`27`	`27`	`for (int64_t m = out_ptr[batch_idx] + 1; m < out_ptr[batch_idx + 1]; m++) {`
	`28`	`+ __syncthreads();`
`28`	`29`	`int64_t old = out[m - 1];`
`29`	`30`
`30`	`31`	`scalar_t best = (scalar_t)-1.;`