Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "62854588d45c9bbaaef7db92724fbdc4c8e5c6a0"
Commit 5f855a28 authored by rusty1s's avatar rusty1s
Browse files

fix illegal memory access

parent 9c33077e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
csrc/cuda/fps_cuda.cu
View file @ 5f855a28
...@@ -25,6 +25,7 @@ __global__ void fps_kernel(const scalar_t *src, const int64_t *ptr, ...@@ -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++) { for (int64_t m = out_ptr[batch_idx] + 1; m < out_ptr[batch_idx + 1]; m++) {
__syncthreads();
int64_t old = out[m - 1]; int64_t old = out[m - 1];
scalar_t best = (scalar_t)-1.; scalar_t best = (scalar_t)-1.;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment