Fix C++ lint (#2059)

9ed2fa3c · Francisco Massa · GitHub · 3c2c0022 · 9ed2fa3c
Unverified Commit 9ed2fa3c authored Apr 03, 2020 by Francisco Massa Committed by GitHub Apr 03, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

torchvision/csrc/cpu/nms_cpu.cpp torchvision/csrc/cpu/nms_cpu.cpp +4 -2

No files found.
--- a/torchvision/csrc/cpu/nms_cpu.cpp
+++ b/torchvision/csrc/cpu/nms_cpu.cpp
@@ -6,9 +6,11 @@ at::Tensor nms_cpu_kernel(
    const at::Tensor& scores,
    const float iou_threshold) {
  AT_ASSERTM(!dets.options().device().is_cuda(), "dets must be a CPU tensor");
-  AT_ASSERTM(!scores.options().device().is_cuda(), "scores must be a CPU tensor");
  AT_ASSERTM(
-      dets.scalar_type() == scores.scalar_type(), "dets should have the same type as scores");
+      !scores.options().device().is_cuda(), "scores must be a CPU tensor");
+  AT_ASSERTM(
+      dets.scalar_type() == scores.scalar_type(),
+      "dets should have the same type as scores");
  if (dets.numel() == 0)
    return at::empty({0}, dets.options().dtype(at::kLong));