diff --git a/pytorch3d/csrc/sample_farthest_points/sample_farthest_points.cu b/pytorch3d/csrc/sample_farthest_points/sample_farthest_points.cu index d9b8df81..70cef75c 100644 --- a/pytorch3d/csrc/sample_farthest_points/sample_farthest_points.cu +++ b/pytorch3d/csrc/sample_farthest_points/sample_farthest_points.cu @@ -155,7 +155,7 @@ at::Tensor FarthestPointSamplingCuda( // Max possible threads per block const int MAX_THREADS_PER_BLOCK = 1024; - const size_t threads = max(min(1 << points_pow_2, MAX_THREADS_PER_BLOCK), 1); + const size_t threads = max(min(1 << points_pow_2, MAX_THREADS_PER_BLOCK), 2); // Create the accessors auto points_a = points.packed_accessor64(); @@ -215,10 +215,6 @@ at::Tensor FarthestPointSamplingCuda( FarthestPointSamplingKernel<2><<>>( points_a, lengths_a, K_a, idxs_a, min_point_dist_a, start_idxs_a); break; - case 1: - FarthestPointSamplingKernel<1><<>>( - points_a, lengths_a, K_a, idxs_a, min_point_dist_a, start_idxs_a); - break; default: FarthestPointSamplingKernel<1024> <<>>(