From f89684d73dad1f863b4fd291e8804b5c24eae94f Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 25 Dec 2020 06:56:46 +0900 Subject: [PATCH] remove ceil_div from nms --- python/tvm/topi/cuda/nms.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) 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.