From 6512a2d6e6c73a8a1cc3b5f7af9f1c13913b233a Mon Sep 17 00:00:00 2001 From: xinlianglalala Date: Tue, 31 Jan 2023 11:50:26 +0800 Subject: [PATCH] amend --- mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp | 3 --- 1 file changed, 3 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp index bbe8970b06d..33887e3fa77 100644 --- a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp @@ -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) @@ -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) @@ -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);