From 56c4af87599a3b5a976b96bb7c6101b0960b713e Mon Sep 17 00:00:00 2001 From: inter Date: Sun, 21 Sep 2025 20:19:22 +0800 Subject: [PATCH] Add File --- pcdet/ops/iou3d_nms/src/iou3d_cpu.h | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 pcdet/ops/iou3d_nms/src/iou3d_cpu.h diff --git a/pcdet/ops/iou3d_nms/src/iou3d_cpu.h b/pcdet/ops/iou3d_nms/src/iou3d_cpu.h new file mode 100644 index 0000000..4d93bb6 --- /dev/null +++ b/pcdet/ops/iou3d_nms/src/iou3d_cpu.h @@ -0,0 +1,11 @@ +#ifndef IOU3D_CPU_H +#define IOU3D_CPU_H + +#include +#include +#include +#include + +int boxes_iou_bev_cpu(at::Tensor boxes_a_tensor, at::Tensor boxes_b_tensor, at::Tensor ans_iou_tensor); +int boxes_aligned_iou_bev_cpu(at::Tensor boxes_a_tensor, at::Tensor boxes_b_tensor, at::Tensor ans_iou_tensor); +#endif