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