From 1a6ad8085686a39a9e62f91aeee3cd2894b6c18f Mon Sep 17 00:00:00 2001 From: xinlianglalala Date: Tue, 31 Jan 2023 14:09:34 +0800 Subject: [PATCH] amend --- mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp index 33887e3fa77..f428311fee0 100644 --- 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;