Skip to content

T
tic
Commit b0a0ae4d by Leyuan Wang Committed by masahi
Options

revert PR#2420 nms changes (#2747)

parent a7e35fc3
Showing with 3 additions and 5 deletions
topi/python/topi/cuda/nms.py
...@@ -35,7 +35,7 @@ def sort_ir(data, index, output): ...@@ -35,7 +35,7 @@ def sort_ir(data, index, output):
p_index = ib.buffer_ptr(index) p_index = ib.buffer_ptr(index)
p_out = ib.buffer_ptr(output) p_out = ib.buffer_ptr(output)
nthread_tx = max_threads nthread_tx = max_threads
nthread_bx = (num_anchors + 1) // 2 // max_threads + 1 nthread_bx = num_anchors // max_threads + 1
tx = tvm.thread_axis("threadIdx.x") tx = tvm.thread_axis("threadIdx.x")
bx = tvm.thread_axis("vthread") bx = tvm.thread_axis("vthread")
ib.scope_attr(tx, "thread_extent", nthread_tx) ib.scope_attr(tx, "thread_extent", nthread_tx)
...@@ -46,10 +46,8 @@ def sort_ir(data, index, output): ...@@ -46,10 +46,8 @@ def sort_ir(data, index, output):
with ib.for_range(0, batch, for_type="unroll") as b: with ib.for_range(0, batch, for_type="unroll") as b:
start = b * num_anchors start = b * num_anchors
for i in range(2): with ib.if_scope(tid < num_anchors):
bbox_id = tid * 2 + i p_out[start + tid] = tid
with ib.if_scope(bbox_id < num_anchors):
p_out[start + bbox_id] = bbox_id
# OddEvenTransposeSort # OddEvenTransposeSort
with ib.for_range(0, p_index[b]) as k: with ib.for_range(0, p_index[b]) as k:
with ib.if_scope(tid < (p_index[b] + 1) // 2): with ib.if_scope(tid < (p_index[b] + 1) // 2):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment