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

---
 pcdet/ops/iou3d_nms/src/iou3d_nms.h | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 pcdet/ops/iou3d_nms/src/iou3d_nms.h

diff --git a/pcdet/ops/iou3d_nms/src/iou3d_nms.h b/pcdet/ops/iou3d_nms/src/iou3d_nms.h
new file mode 100644
index 0000000..3202027
--- /dev/null
+++ b/pcdet/ops/iou3d_nms/src/iou3d_nms.h
@@ -0,0 +1,17 @@
+#ifndef IOU3D_NMS_H
+#define IOU3D_NMS_H
+
+#include <torch/serialize/tensor.h>
+#include <vector>
+#include <assert.h>
+#include <cuda.h>
+#include <cuda_runtime_api.h>
+
+int boxes_aligned_overlap_bev_gpu(at::Tensor boxes_a, at::Tensor boxes_b, at::Tensor ans_overlap);
+int boxes_overlap_bev_gpu(at::Tensor boxes_a, at::Tensor boxes_b, at::Tensor ans_overlap);
+int paired_boxes_overlap_bev_gpu(at::Tensor boxes_a, at::Tensor boxes_b, at::Tensor ans_overlap);
+int boxes_iou_bev_gpu(at::Tensor boxes_a, at::Tensor boxes_b, at::Tensor ans_iou);
+int nms_gpu(at::Tensor boxes, at::Tensor keep, float nms_overlap_thresh);
+int nms_normal_gpu(at::Tensor boxes, at::Tensor keep, float nms_overlap_thresh);
+
+#endif