From bfdd1f9c4281316f4f654672f0e9dbe1b0624ff0 Mon Sep 17 00:00:00 2001 From: sherie <963372609@qq.com> Date: Tue, 4 Apr 2023 16:05:42 +0800 Subject: [PATCH] [Fix] Fix boxes.scalar_type in nms_npu (#2731) --- mmcv/ops/csrc/pytorch/npu/nms_npu.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp b/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp index e5a5ae780..0a1f997a2 100644 --- a/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/nms_npu.cpp @@ -4,7 +4,7 @@ using namespace NPU_NAME_SPACE; using namespace std; Tensor nms_npu(Tensor boxes, Tensor scores, float iou_threshold, int offset) { - TORCH_CHECK((boxes.scalar_type == at::ScalarType::Float), + TORCH_CHECK((boxes.scalar_type() == at::ScalarType::Float), "The type of boxes tensor passed in nms_npu should be float"); int64_t offset_64 = offset; at::Tensor iou_threshold_y = at_npu::native::OpPreparation::ApplyTensor(