From e1854b43e066e87bd152baebeddc018cdc9e1a0f Mon Sep 17 00:00:00 2001
From: inter <inter@qq.com>
Date: Sun, 21 Sep 2025 20:19:18 +0800
Subject: [PATCH] Add File

---
 .../pointnet2_batch/src/pointnet2_api.cpp     | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 pcdet/ops/pointnet2/pointnet2_batch/src/pointnet2_api.cpp

diff --git a/pcdet/ops/pointnet2/pointnet2_batch/src/pointnet2_api.cpp b/pcdet/ops/pointnet2/pointnet2_batch/src/pointnet2_api.cpp
new file mode 100644
index 0000000..2843650
--- /dev/null
+++ b/pcdet/ops/pointnet2/pointnet2_batch/src/pointnet2_api.cpp
@@ -0,0 +1,24 @@
+#include <torch/serialize/tensor.h>
+#include <torch/extension.h>
+
+#include "ball_query_gpu.h"
+#include "group_points_gpu.h"
+#include "sampling_gpu.h"
+#include "interpolate_gpu.h"
+
+
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
+    m.def("ball_query_wrapper", &ball_query_wrapper_fast, "ball_query_wrapper_fast");
+
+    m.def("group_points_wrapper", &group_points_wrapper_fast, "group_points_wrapper_fast");
+    m.def("group_points_grad_wrapper", &group_points_grad_wrapper_fast, "group_points_grad_wrapper_fast");
+
+    m.def("gather_points_wrapper", &gather_points_wrapper_fast, "gather_points_wrapper_fast");
+    m.def("gather_points_grad_wrapper", &gather_points_grad_wrapper_fast, "gather_points_grad_wrapper_fast");
+
+    m.def("farthest_point_sampling_wrapper", &farthest_point_sampling_wrapper, "farthest_point_sampling_wrapper");
+    
+    m.def("three_nn_wrapper", &three_nn_wrapper_fast, "three_nn_wrapper_fast");
+    m.def("three_interpolate_wrapper", &three_interpolate_wrapper_fast, "three_interpolate_wrapper_fast");
+    m.def("three_interpolate_grad_wrapper", &three_interpolate_grad_wrapper_fast, "three_interpolate_grad_wrapper_fast");
+}