Skip to content

Commit

Permalink
[Fix] Fix the parameter error of DeformableRoiPool in Ascend device (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
dflhw authored Jan 19, 2023
1 parent dfb0380 commit 0b005c5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -45,11 +45,11 @@ void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tensor rois,
int64_t sampling_ratio_ = (int64_t)sampling_ratio;
OpCommand cmd;
cmd.Name("DeformableRoiPoolGrad")
.Input(grad_input)
.Input(grad_output)
.Input(input)
.Input(rois)
.Input(offset)
.Output(grad_output)
.Output(grad_input)
.Output(grad_offset)
.Attr("output_size", output_size)
.Attr("spatial_scale", spatial_scale)
Expand Down

0 comments on commit 0b005c5

Please sign in to comment.