fix number of threads which would compute the final mask

prime-slam · Apr 14, 2024 · 132f54a · 132f54a
1 parent 37d554a
commit 132f54a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/octreelib/ransac/cuda_ransac.py b/octreelib/ransac/cuda_ransac.py
@@ -149,7 +149,7 @@ def kernel(
             for i in range(
                 block_start_indices[block_id] + thread_id,
                 block_start_indices[block_id] + block_sizes[block_id],
-                CUDA_THREADS,
+                cuda.blockDim.x,
             ):
                 if measure_distance(best_plane, point_cloud[i]) < threshold:
                     result_mask[i] = True