fix torch

open-mmlab · Oct 19, 2023 · cf1c154 · cf1c154
1 parent 6123af4
commit cf1c154
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp b/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp
@@ -111,9 +111,8 @@ void softmax_focal_loss_forward_npu(Tensor input, Tensor target, Tensor weight,
       .Input(op_output)
       .Input(offsetVec)
       .Input(sizeVec)
-      .Output(result)
+      .Output(output)
       .Run();
-  return result;
 }
 
 void softmax_focal_loss_forward_impl(Tensor input, Tensor target, Tensor weight,