From c38593d25ec5bab75c4b4c8ad8ca4ac83719ac1a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=83=A1=E5=AE=8F=E7=AC=8B?= <huhongsun1@huawei.com>
Date: Fri, 14 Jun 2024 17:11:39 +0800
Subject: [PATCH] repair nms_rotated bug

---
 mmcv/ops/nms.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mmcv/ops/nms.py b/mmcv/ops/nms.py
index 5115a95f62..d0c761ce39 100644
--- a/mmcv/ops/nms.py
+++ b/mmcv/ops/nms.py
@@ -452,7 +452,7 @@ def nms_rotated(dets: Tensor,
         flip_mat[-1] = -1
         dets_cw = dets * flip_mat
     else:
-        dets_cw = dets
+        dets_cw = dets.clone()
     multi_label = labels is not None
     if labels is None:
         input_labels = scores.new_empty(0, dtype=torch.int)
@@ -462,6 +462,8 @@ def nms_rotated(dets: Tensor,
         order = scores.new_empty(0, dtype=torch.long)
         if dets.device.type == 'npu':
             coefficient = 57.29578  # 180 / PI
+            dets_cw = dets_cw.float()
+            scores = scores.float()
             for i in range(dets.size()[0]):
                 dets_cw[i][4] *= coefficient  # radians to angle
         keep_inds = ext_module.nms_rotated(dets_cw, scores, order, dets_cw,