fix bug

hust17yixuan · Feb 6, 2025 · 727f2ca · 727f2ca
1 parent 0fae27c
commit 727f2ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp b/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp
@@ -13,7 +13,7 @@ Tensor nms_npu(Tensor boxes, Tensor scores, float iou_threshold, int offset) {
       at::empty({}, boxes.options().dtype(at::kFloat)).fill_(0);
   at::Tensor max_outputsize_y =
       at::empty({}, boxes.options().dtype(at::kInt)).fill_(boxes.size(0));
-  c10::SmallVector<int64_t, SIZE> outputsize = {boxes.size(0)};
+  c10::SmallVector<int64_t, 8> outputsize = {boxes.size(0)};
   at::Tensor output =
       at::empty(outputsize, boxes.options().dtype(at::kInt)).fill_(-1);
   OpCommand cmd;