diff --git a/mmcv/ops/csrc/pytorch/npu/bbox_overlaps_npu.cpp b/mmcv/ops/csrc/pytorch/npu/bbox_overlaps_npu.cpp index ed04622af..3e346e733 100644 --- a/mmcv/ops/csrc/pytorch/npu/bbox_overlaps_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/bbox_overlaps_npu.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE; diff --git a/mmcv/ops/csrc/pytorch/npu/common_util.h b/mmcv/ops/csrc/pytorch/npu/common_util.h index db05e6a05..209276386 100644 --- a/mmcv/ops/csrc/pytorch/npu/common_util.h +++ b/mmcv/ops/csrc/pytorch/npu/common_util.h @@ -2,11 +2,4 @@ #define MMCV_OPS_CSRC_COMMON__UTIL_HPP_ const int SIZE = 8; -c10::SmallVector array_to_vector(c10::IntArrayRef shape) { - c10::SmallVector shape_small_vec; - for (uint64_t i = 0; i < shape.size(); i++) { - shape_small_vec.emplace_back(shape[i]); - } -} - #endif // MMCV_OPS_CSRC_COMMON__UTIL_HPP_ diff --git a/mmcv/ops/csrc/pytorch/npu/fused_bias_leakyrelu_npu.cpp b/mmcv/ops/csrc/pytorch/npu/fused_bias_leakyrelu_npu.cpp index 2e1270e45..e8cd552ef 100644 --- a/mmcv/ops/csrc/pytorch/npu/fused_bias_leakyrelu_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/fused_bias_leakyrelu_npu.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE; diff --git a/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp b/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp index 5638c8be3..a7fbebee9 100644 --- a/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE; diff --git a/mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp b/mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp index 717f954cf..1b2107eb9 100644 --- a/mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE; diff --git a/mmcv/ops/csrc/pytorch/npu/roi_align_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_align_npu.cpp index 0e673614f..b1edd8886 100644 --- a/mmcv/ops/csrc/pytorch/npu/roi_align_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/roi_align_npu.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE; diff --git a/mmcv/ops/csrc/pytorch/npu/roi_align_rotated_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_align_rotated_npu.cpp index 2a3ff09e9..3d7560f49 100644 --- a/mmcv/ops/csrc/pytorch/npu/roi_align_rotated_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/roi_align_rotated_npu.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE; diff --git a/mmcv/ops/csrc/pytorch/npu/roipoint_pool3d_forward.cpp b/mmcv/ops/csrc/pytorch/npu/roipoint_pool3d_forward.cpp index 48d5b4789..1cf6500ee 100644 --- a/mmcv/ops/csrc/pytorch/npu/roipoint_pool3d_forward.cpp +++ b/mmcv/ops/csrc/pytorch/npu/roipoint_pool3d_forward.cpp @@ -1,3 +1,4 @@ +#include "common_util.h" #include "pytorch_npu_helper.hpp" using namespace NPU_NAME_SPACE;