Skip to content

Commit

Permalink
amend
Browse files Browse the repository at this point in the history
  • Loading branch information
xinlianglalala committed Jan 31, 2023
1 parent 91428d6 commit 6512a2d
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ void roi_pool_forward_npu(Tensor input, Tensor rois, Tensor output,
int64_t pooled_channel = 1;
at::Tensor roi_actual_num = at_npu::native::OpPreparation::ApplyTensor(
{}, rois.options().dtype(at::kInt), rois);

OpCommand cmd;
cmd.Name("RoiPoolingWithArgMax")
.Input(input)
Expand All @@ -36,7 +35,6 @@ void roi_pool_backward_npu(Tensor grad_output, Tensor rois, Tensor argmax,
at::Tensor roi_actual_num = at_npu::native::OpPreparation::ApplyTensor(
{}, rois.options().dtype(at::kInt), rois);
at::Tensor x = at::ones_like(grad_input);

OpCommand cmd;
cmd.Name("RoiPoolingGradWithArgMax")
.Input(grad_output)
Expand All @@ -62,5 +60,4 @@ void roi_pool_backward_impl(Tensor grad_output, Tensor rois, Tensor argmax,
int pooled_width, float spatial_scale);

REGISTER_NPU_IMPL(roi_pool_forward_impl, roi_pool_forward_npu);

REGISTER_NPU_IMPL(roi_pool_backward_impl, roi_pool_backward_npu);

0 comments on commit 6512a2d

Please sign in to comment.