diff --git a/python/tvm/topi/cuda/nms.py b/python/tvm/topi/cuda/nms.py index f08389faf77c..4b6452918a4a 100644 --- 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.