Updated max value for tmp in furthest sampler to allow support for ha…

…lf type
mihdalal · Apr 10, 2021 · c03ead5 · c03ead5
1 parent dffc0f3
commit c03ead5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pointnet2_ops/_ext-src/src/sampling_gpu.cu b/pointnet2_ops/_ext-src/src/sampling_gpu.cu
@@ -228,7 +228,7 @@ at::Tensor furthest_point_sampling_kernel_wrapper(
                    at::device(points.device()).dtype(at::ScalarType::Int));
 
   at::Tensor tmp =
-      torch::full({points.size(0), points.size(1)}, 1e10,
+      torch::full({points.size(0), points.size(1)}, 6e5,
                   at::device(points.device()).dtype(points.scalar_type()));
 
   AT_DISPATCH_FLOATING_TYPES_AND_HALF(