fix torch_npu adpater

momo609 · momo609 · commit 6919f48270c0 · 2023-11-07T09:38:58.000+08:00
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;

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,7 @@`
`26`	`26`
`27`	`27`	`#define NPU_NAME_SPACE at_npu::native`
`28`	`28`
	`29`	`+const int SIZE = 8;`
`29`	`30`	`c10::SmallVector<int64_t, SIZE> array_to_small_vector(c10::IntArrayRef shape)`
`30`	`31`	`{`
`31`	`32`	`c10::SmallVector<int64_t, SIZE> shape_small_vec;`