fix bug.

momo609 · May 30, 2024 · 97715d6 · 97715d6
1 parent 50a3552
commit 97715d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp
@@ -70,7 +70,7 @@ void roi_pool_backward_npu(Tensor grad_output, Tensor rois, Tensor argmax,
       .Attr("spatial_scale_w", spatial_scale)
       .Attr("pool_channel", pooled_channel)
       .Run();
-  at::Tensor res = NpuUtils::format_contiguous(result);
+  at::Tensor res = NpuUtils::format_contiguous(y);
   grad_input.copy_(res);
 }