add nms_cuda.c

Author: Jianwei Yang

lib/model/nms/src/nms_cuda.c

@@ -0,0 +1,19 @@
+#include <THC/THC.h>
+#include <stdio.h>
+#include "nms_cuda_kernel.h"
+
+// this symbol will be resolved automatically from PyTorch libs
+extern THCState *state;
+
+int nms_cuda(THCudaIntTensor *keep_out, THCudaTensor *boxes_host,
+		     THCudaIntTensor *num_out, float nms_overlap_thresh) {
+
+	nms_cuda_compute(THCudaIntTensor_data(state, keep_out), 
+		         THCudaIntTensor_data(state, num_out), 
+      	                 THCudaTensor_data(state, boxes_host), 
+		         boxes_host->size[0], 
+		         boxes_host->size[1],
+		         nms_overlap_thresh);
+
+	return 1;
+}