fix torch

open-mmlab · Oct 19, 2023 · 6123af4 · 6123af4
1 parent 2ceeded
commit 6123af4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp b/mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp
@@ -108,7 +108,7 @@ void softmax_focal_loss_forward_npu(Tensor input, Tensor target, Tensor weight,
   c10::SmallVector<int64_t, N> offsetVec = array_to_small_vector(offset);
   c10::SmallVector<int64_t, N> sizeVec = array_to_small_vector(size_array);
   cmd.Name("Slice")
-      .Input(self)
+      .Input(op_output)
       .Input(offsetVec)
       .Input(sizeVec)
       .Output(result)