fix illegal memory access

5f855a28 · rusty1s · 9c33077e · 5f855a28
Commit 5f855a28 authored Mar 26, 2020 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

csrc/cuda/fps_cuda.cu csrc/cuda/fps_cuda.cu +1 -0

No files found.
--- 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.;