From 6919f48270c0f5bd2175ae19bf2bfec2acf49283 Mon Sep 17 00:00:00 2001 From: momo609 <963372609@qq.com> Date: Tue, 7 Nov 2023 09:38:58 +0800 Subject: [PATCH] fix torch_npu adpater --- mmcv/ops/csrc/common/pytorch_npu_helper.hpp | 1 + 1 file changed, 1 insertion(+) diff --git a/mmcv/ops/csrc/common/pytorch_npu_helper.hpp b/mmcv/ops/csrc/common/pytorch_npu_helper.hpp index 918b2e01f85..01cfe805488 100644 --- a/mmcv/ops/csrc/common/pytorch_npu_helper.hpp +++ b/mmcv/ops/csrc/common/pytorch_npu_helper.hpp @@ -26,6 +26,7 @@ #define NPU_NAME_SPACE at_npu::native +const int SIZE = 8; c10::SmallVector array_to_small_vector(c10::IntArrayRef shape) { c10::SmallVector shape_small_vec;