From e5a372646c53f19111322474d625c7a906af23bd Mon Sep 17 00:00:00 2001
From: momo609 <963372609@qq.com>
Date: Tue, 26 Sep 2023 10:24:15 +0800
Subject: [PATCH] fix nms_rotated

---
 mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp b/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp
index a6e08f1067a..b7c995a223e 100644
--- a/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp
+++ b/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp
@@ -53,8 +53,7 @@ void sigmoid_focal_loss_backward_npu(Tensor input, Tensor target, Tensor weight,
   int64_t weight_size = weight.size(0);
   at::Tensor weight_y = at::ones_like(input);
   if (weight_size > 0) {
-    weight_y = at_npu::native::NPUNativeFunctions::npu_broadcast(weight,
-                                                                 input.sizes());
+    weight_y = at::broadcast_to(weight, input.sizes());
   }
   OpCommand cmd;
   string reduction = "none";