Skip to content

Commit 9ed2fa3

Browse files
authored
Fix C++ lint (#2059)
1 parent 3c2c002 commit 9ed2fa3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torchvision/csrc/cpu/nms_cpu.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,11 @@ at::Tensor nms_cpu_kernel(
66
const at::Tensor& scores,
77
const float iou_threshold) {
88
AT_ASSERTM(!dets.options().device().is_cuda(), "dets must be a CPU tensor");
9-
AT_ASSERTM(!scores.options().device().is_cuda(), "scores must be a CPU tensor");
109
AT_ASSERTM(
11-
dets.scalar_type() == scores.scalar_type(), "dets should have the same type as scores");
10+
!scores.options().device().is_cuda(), "scores must be a CPU tensor");
11+
AT_ASSERTM(
12+
dets.scalar_type() == scores.scalar_type(),
13+
"dets should have the same type as scores");
1214

1315
if (dets.numel() == 0)
1416
return at::empty({0}, dets.options().dtype(at::kLong));

0 commit comments

Comments
 (0)