amend

open-mmlab · Jan 31, 2023 · 1a6ad80 · 1a6ad80
1 parent 6512a2d
commit 1a6ad80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp
@@ -27,8 +27,8 @@ void roi_pool_forward_npu(Tensor input, Tensor rois, Tensor output,
 }
 
 void roi_pool_backward_npu(Tensor grad_output, Tensor rois, Tensor argmax,
-                            Tensor grad_input, int pooled_height,
-                            int pooled_width, float spatial_scale) {
+                           Tensor grad_input, int pooled_height,
+                           int pooled_width, float spatial_scale) {
   int64_t pooled_height_64 = pooled_height;
   int64_t pooled_width_64 = pooled_width;
   int64_t pooled_channel = 1;