From ab33796864dcb857ec1eaca677737d6b5d787a6b Mon Sep 17 00:00:00 2001
From: Mikhail Kiselyov <64229743+true-real-michael@users.noreply.github.com>
Date: Sun, 14 Apr 2024 12:50:08 +0300
Subject: [PATCH] fix number of threads which would compute the final mask
 (#31)

---
 octreelib/ransac/cuda_ransac.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/octreelib/ransac/cuda_ransac.py b/octreelib/ransac/cuda_ransac.py
index 9293b2f..ee164c4 100644
--- 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