We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bc727f7 commit bfdd1f9Copy full SHA for bfdd1f9
mmcv/ops/csrc/pytorch/npu/nms_npu.cpp
@@ -4,7 +4,7 @@ using namespace NPU_NAME_SPACE;
4
using namespace std;
5
6
Tensor nms_npu(Tensor boxes, Tensor scores, float iou_threshold, int offset) {
7
- TORCH_CHECK((boxes.scalar_type == at::ScalarType::Float),
+ TORCH_CHECK((boxes.scalar_type() == at::ScalarType::Float),
8
"The type of boxes tensor passed in nms_npu should be float");
9
int64_t offset_64 = offset;
10
at::Tensor iou_threshold_y = at_npu::native::OpPreparation::ApplyTensor(
0 commit comments