brute_force_knn.cuh 451 B

1234567891011121314151617
  1. #ifndef FAST_GICP_CUDA_BRUTE_FORCE_KNN_CUH
  2. #define FAST_GICP_CUDA_BRUTE_FORCE_KNN_CUH
  3. #include <Eigen/Core>
  4. #include <thrust/device_vector.h>
  5. namespace fast_gicp {
  6. namespace cuda {
  7. void brute_force_knn_search(const thrust::device_vector<Eigen::Vector3f>& source, const thrust::device_vector<Eigen::Vector3f>& target, int k, thrust::device_vector<thrust::pair<float, int>>& k_neighbors, bool do_sort=false);
  8. }
  9. } // namespace fast_gicp
  10. #endif