diff --git a/mmcv/ops/csrc/pytorch/nms_rotated.cpp b/mmcv/ops/csrc/pytorch/nms_rotated.cpp index 3cbcb0b09ba..ed669169a3f 100644 --- a/mmcv/ops/csrc/pytorch/nms_rotated.cpp +++ b/mmcv/ops/csrc/pytorch/nms_rotated.cpp @@ -31,7 +31,7 @@ Tensor nms_rotated(const Tensor dets, const Tensor scores, const Tensor order, #else AT_ERROR("Not compiled with GPU support"); #endif - } else if(dets.device().type() == at::kXLA) { + } else if (dets.device().type() == at::kXLA) { #ifdef MMCV_WITH_NPU return nms_rotated_npu(dets, scores, labels, iou_threshold); #else