diff --git a/mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp b/mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp index 2138b2f8b..614f1ea8c 100644 --- a/mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp @@ -41,5 +41,5 @@ void assign_score_withk_backward_impl( const Tensor& scores, const Tensor& knn_idx, Tensor& grad_points, Tensor& grad_centers, Tensor& grad_scores); -REGISTER_NPU_IMPL(assign_score_withk_backward_impl, +REGISTER_NPU_IMPL(assign_score_withk_backward_impl, assign_score_withk_backward_npu);