Skip to content

Commit 12a3dd8

Browse files
committed
fix torch_npu adpater
1 parent 66b6a15 commit 12a3dd8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mmcv/ops/csrc/common/pytorch_npu_helper.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626

2727
#define NPU_NAME_SPACE at_npu::native
2828

29+
const int SIZE = 8;
2930
c10::SmallVector<int64_t, SIZE> array_to_small_vector(c10::IntArrayRef shape)
3031
{
3132
c10::SmallVector<int64_t, SIZE> shape_small_vec;

0 commit comments

Comments
 (0)