remove ceil_div from nms

masahi · Dec 24, 2020 · f89684d · f89684d
1 parent a2ad4de
commit f89684d
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/tvm/topi/cuda/nms.py b/python/tvm/topi/cuda/nms.py
@@ -23,6 +23,7 @@
 from tvm.tir import if_then_else
 from .sort import argsort, argsort_thrust
 from .prefix_scan import exclusive_scan
+from ..utils import ceil_div
 
 
 def cuda_atomic_add_rule(op):
@@ -52,10 +53,6 @@ def atomic_add(x, y):
     return tvm.tir.call_intrin(y.dtype, "tir.atomic_add", x, y)
 
 
-def ceil_div(a, b):
-    return tvm.tir.indexdiv(a + b - 1, b)
-
-
 def get_valid_boxes_ir(data, valid_boxes, score_threshold, id_index, score_index):
     """Low level IR to identify bounding boxes given a score threshold.