diff --git a/mmcv/ops/csrc/pytorch/nms_rotated.cpp b/mmcv/ops/csrc/pytorch/nms_rotated.cpp index e4ef676a9d..b633983053 100644 --- a/mmcv/ops/csrc/pytorch/nms_rotated.cpp +++ b/mmcv/ops/csrc/pytorch/nms_rotated.cpp @@ -21,12 +21,12 @@ Tensor nms_rotated(const Tensor dets, const Tensor scores, const Tensor order, assert(dets.device().is_cuda() == scores.device().is_cuda()); if (dets.device().is_cuda()) { #ifdef MMCV_WITH_CUDA - return nms_rotated_cuda(dets, scores, order, dets_sorted, iou_threshold, - multi_label); + return nms_rotated_cuda(dets, scores, order, dets_sorted.contiguous(), + iou_threshold, multi_label); #else AT_ERROR("Not compiled with GPU support"); #endif } - return nms_rotated_cpu(dets, scores, iou_threshold); + return nms_rotated_cpu(dets.contiguous(), scores.contiguous(), iou_threshold); }