Skip to content

Commit 2e523a0

Browse files
committed
fix torch
1 parent 561388a commit 2e523a0

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

mmcv/ops/csrc/pytorch/npu/roi_align_rotated_npu.cpp

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@ using namespace NPU_NAME_SPACE;
44
using namespace std;
55

66
void roi_align_rotated_forward_npu(Tensor input, Tensor rois, Tensor output,
7-
int aligned_height, int aligned_width,
8-
float spatial_scale, int sampling_ratio,
9-
bool aligned, bool clockwise) {
7+
int aligned_height, int aligned_width,
8+
float spatial_scale, int sampling_ratio,
9+
bool aligned, bool clockwise) {
1010
int64_t aligned_height_64 = aligned_height;
1111
int64_t aligned_width_64 = aligned_width;
1212
int64_t sampling_ratio_64 = sampling_ratio;
@@ -25,10 +25,10 @@ void roi_align_rotated_forward_npu(Tensor input, Tensor rois, Tensor output,
2525
}
2626

2727
void roi_align_rotated_backward_npu(Tensor top_grad, Tensor rois,
28-
Tensor bottom_grad, int aligned_height,
29-
int aligned_width, float spatial_scale,
30-
int sampling_ratio, bool aligned,
31-
bool clockwise) {
28+
Tensor bottom_grad, int aligned_height,
29+
int aligned_width, float spatial_scale,
30+
int sampling_ratio, bool aligned,
31+
bool clockwise) {
3232
int64_t aligned_height_64 = aligned_height;
3333
int64_t aligned_width_64 = aligned_width;
3434
int64_t sampling_ratio_64 = sampling_ratio;
@@ -60,5 +60,7 @@ void roi_align_rotated_backward_impl(Tensor top_grad, Tensor rois,
6060
int sampling_ratio, bool aligned,
6161
bool clockwise);
6262

63-
REGISTER_NPU_IMPL(roi_align_rotated_forward_impl, roi_align_rotated_forward_npu);
64-
REGISTER_NPU_IMPL(roi_align_rotated_backward_impl, roi_align_rotated_backward_npu);
63+
REGISTER_NPU_IMPL(roi_align_rotated_forward_impl,
64+
roi_align_rotated_forward_npu);
65+
REGISTER_NPU_IMPL(roi_align_rotated_backward_impl,
66+
roi_align_rotated_backward_npu);

0 commit comments

Comments
 (0)