#include #include #include #include "sampling_cuda_kernel.h" void farthest_point_sampling_cuda(int b, int n, at::Tensor xyz_tensor, at::Tensor offset_tensor, at::Tensor new_offset_tensor, at::Tensor tmp_tensor, at::Tensor idx_tensor) { const float *xyz = xyz_tensor.data_ptr(); const int *offset = offset_tensor.data_ptr(); const int *new_offset = new_offset_tensor.data_ptr(); float *tmp = tmp_tensor.data_ptr(); int *idx = idx_tensor.data_ptr(); farthest_point_sampling_cuda_launcher(b, n, xyz, offset, new_offset, tmp, idx); }