Function PYBIND11_MODULE

pointersect/pr/cuda/pr_cuda.cpp:734–747

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    m.def("get_grid_idx", &get_grid_idx, "compute the grid index for each point");
    m.def("gather_points_v1", &gather_points_v1, "gather the points in each cell");
    m.def("gather_points_v2", &gather_points_v2, "gather the points in each cell");
    m.def("grid_ray_intersection", &grid_ray_intersection, "intersect ray with a grid");
    m.def("collect_points_on_ray", &collect_points_on_ray, "collect points along a ray using compiled structure");
    m.def("find_neighbor_points_of_rays", &find_neighbor_points_of_rays, "find the neighboring points of a ray");
    m.def("find_k_neighbor_points_of_rays", &find_k_neighbor_points_of_rays, "find the k_neighboring points of a ray");
    m.def("keep_min_k_values", &keep_min_k_values, "find k-min values with max heap");
    m.def("find_k_neighbor_points_of_rays_v2", &find_k_neighbor_points_of_rays_v2, "find the k_neighboring points of a ray (version 2)");
    m.def("find_k_neighbor_points_of_rays_v3", &find_k_neighbor_points_of_rays_v3, "find the k_neighboring points of a ray (version 3)");
    m.def("find_k_neighbor_points_of_rays_v4", &find_k_neighbor_points_of_rays_v4, "find the k_neighboring points of a ray (version 4)");
}