1.4.0 patch (#246)

90c6a951 · Hang Zhang · GitHub · ae218bf9 · 90c6a951 · 90c6a951
Unverified Commit 90c6a951 authored Mar 15, 2020 by Hang Zhang Committed by GitHub Mar 15, 2020
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

encoding/lib/cpu/nms_cpu.cpp encoding/lib/cpu/nms_cpu.cpp +2 -2

encoding/lib/gpu/nms_kernel.cu encoding/lib/gpu/nms_kernel.cu +2 -2

No files found.
--- 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());


--- 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);