From 132f54a57b76f7d7f68a98fcec23e4e24be94ca8 Mon Sep 17 00:00:00 2001
From: Mikhail Kiselyov <kiselev.0353@gmail.com>
Date: Sun, 14 Apr 2024 12:46:24 +0300
Subject: [PATCH] fix number of threads which would compute the final mask

---
 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