From 7d560cf43e3b5d85b5c539dfa286f8b9c8e9fabe Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Fri, 3 Apr 2020 12:07:31 +0200 Subject: [PATCH] Fix C++ lint --- torchvision/csrc/cpu/nms_cpu.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/torchvision/csrc/cpu/nms_cpu.cpp b/torchvision/csrc/cpu/nms_cpu.cpp index 4549b39b1b4..199c469bff8 100644 --- 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));