fix torch_npu adpater

open-mmlab · Nov 7, 2023 · 6919f48 · 6919f48
1 parent 1b55643
commit 6919f48
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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<int64_t, SIZE> array_to_small_vector(c10::IntArrayRef shape)
 {
   c10::SmallVector<int64_t, SIZE> shape_small_vec;