From 90c6a9517a565fa4276ae1b58ac0a73f739cc3f4 Mon Sep 17 00:00:00 2001
From: Hang Zhang <zhang.hang@rutgers.edu>
Date: Sun, 15 Mar 2020 18:53:48 -0700
Subject: [PATCH] 1.4.0 patch (#246)

---
 encoding/lib/cpu/nms_cpu.cpp   | 4 ++--
 encoding/lib/gpu/nms_kernel.cu | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/encoding/lib/cpu/nms_cpu.cpp b/encoding/lib/cpu/nms_cpu.cpp
index d078f30e..b04446f7 100644
--- a/encoding/lib/cpu/nms_cpu.cpp
+++ b/encoding/lib/cpu/nms_cpu.cpp
@@ -32,8 +32,8 @@ std::vector<at::Tensor> Non_Max_Suppression_CPU(
   AT_ASSERT(input.size(2) == 4);
   AT_ASSERT(input.is_contiguous());
   AT_ASSERT(scores.is_contiguous());
-  AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble)
-  AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble)
+  AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble);
+  AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble);
   AT_ASSERT(input.is_contiguous());
   AT_ASSERT(scores.is_contiguous());
 
diff --git a/encoding/lib/gpu/nms_kernel.cu b/encoding/lib/gpu/nms_kernel.cu
index 9c350a7f..b57353c5 100644
--- a/encoding/lib/gpu/nms_kernel.cu
+++ b/encoding/lib/gpu/nms_kernel.cu
@@ -71,8 +71,8 @@ std::vector<at::Tensor> Non_Max_Suppression_CUDA(
   AT_ASSERT(input.size(2) == 4);
   AT_ASSERT(input.is_contiguous());
   AT_ASSERT(scores.is_contiguous());
-  AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble)
-  AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble)
+  AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble);
+  AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble);
 
   auto num_boxes = input.size(1);
   auto batch_size = input.size(0);