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

---
 pcdet/ops/ingroup_inds/src/ingroup_inds.cpp | 54 +++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 pcdet/ops/ingroup_inds/src/ingroup_inds.cpp

diff --git a/pcdet/ops/ingroup_inds/src/ingroup_inds.cpp b/pcdet/ops/ingroup_inds/src/ingroup_inds.cpp
new file mode 100644
index 0000000..8bd3389
--- /dev/null
+++ b/pcdet/ops/ingroup_inds/src/ingroup_inds.cpp
@@ -0,0 +1,54 @@
+#include <assert.h>
+#include <torch/extension.h>
+#include <torch/serialize/tensor.h>
+#include <vector>
+
+#define CHECK_CUDA(x) \
+  TORCH_CHECK(x.device().is_cuda(), #x, " must be a CUDAtensor ")
+#define CHECK_CONTIGUOUS(x) \
+  TORCH_CHECK(x.is_contiguous(), #x, " must be contiguous ")
+#define CHECK_INPUT(x) \
+  CHECK_CUDA(x);       \
+  CHECK_CONTIGUOUS(x)
+
+
+void ingroup_inds_launcher(
+    const long *group_inds_data,
+    long *out_inds_data,
+    int N,
+    int max_group_id
+);
+
+
+void ingroup_inds_gpu(
+  at::Tensor group_inds,
+  at::Tensor out_inds
+);
+
+void ingroup_inds_gpu(
+  at::Tensor group_inds,
+  at::Tensor out_inds
+) {
+
+  CHECK_INPUT(group_inds);
+  CHECK_INPUT(out_inds);
+  int N = group_inds.size(0);
+  int max_group_id = group_inds.max().item().toLong();
+
+
+  long *group_inds_data = group_inds.data_ptr<long>();
+  long *out_inds_data = out_inds.data_ptr<long>();
+
+  ingroup_inds_launcher(
+      group_inds_data,
+      out_inds_data,
+      N,
+      max_group_id
+  );
+
+}
+
+
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
+  m.def("forward", &ingroup_inds_gpu, "cuda version of get_inner_win_inds of SST");
+}
\ No newline at end of file