From 0bed8e2e839125c0ecf8b271c76cbdc81d4cecda Mon Sep 17 00:00:00 2001 From: inter Date: Sun, 21 Sep 2025 20:19:12 +0800 Subject: [PATCH] Add File --- .../pointnet2_stack/src/pointnet2_api.cpp | 31 +++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 pcdet/ops/pointnet2/pointnet2_stack/src/pointnet2_api.cpp diff --git a/pcdet/ops/pointnet2/pointnet2_stack/src/pointnet2_api.cpp b/pcdet/ops/pointnet2/pointnet2_stack/src/pointnet2_api.cpp new file mode 100644 index 0000000..1b61e41 --- /dev/null +++ b/pcdet/ops/pointnet2/pointnet2_stack/src/pointnet2_api.cpp @@ -0,0 +1,31 @@ +#include +#include + +#include "ball_query_gpu.h" +#include "group_points_gpu.h" +#include "sampling_gpu.h" +#include "interpolate_gpu.h" +#include "voxel_query_gpu.h" +#include "vector_pool_gpu.h" + + +PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { + m.def("ball_query_wrapper", &ball_query_wrapper_stack, "ball_query_wrapper_stack"); + m.def("voxel_query_wrapper", &voxel_query_wrapper_stack, "voxel_query_wrapper_stack"); + + m.def("farthest_point_sampling_wrapper", &farthest_point_sampling_wrapper, "farthest_point_sampling_wrapper"); + m.def("stack_farthest_point_sampling_wrapper", &stack_farthest_point_sampling_wrapper, "stack_farthest_point_sampling_wrapper"); + + m.def("group_points_wrapper", &group_points_wrapper_stack, "group_points_wrapper_stack"); + m.def("group_points_grad_wrapper", &group_points_grad_wrapper_stack, "group_points_grad_wrapper_stack"); + + m.def("three_nn_wrapper", &three_nn_wrapper_stack, "three_nn_wrapper_stack"); + m.def("three_interpolate_wrapper", &three_interpolate_wrapper_stack, "three_interpolate_wrapper_stack"); + m.def("three_interpolate_grad_wrapper", &three_interpolate_grad_wrapper_stack, "three_interpolate_grad_wrapper_stack"); + + m.def("query_stacked_local_neighbor_idxs_wrapper_stack", &query_stacked_local_neighbor_idxs_wrapper_stack, "query_stacked_local_neighbor_idxs_wrapper_stack"); + m.def("query_three_nn_by_stacked_local_idxs_wrapper_stack", &query_three_nn_by_stacked_local_idxs_wrapper_stack, "query_three_nn_by_stacked_local_idxs_wrapper_stack"); + + m.def("vector_pool_wrapper", &vector_pool_wrapper_stack, "vector_pool_grad_wrapper_stack"); + m.def("vector_pool_grad_wrapper", &vector_pool_grad_wrapper_stack, "vector_pool_grad_wrapper_stack"); +}