Skip to content

Commit

Permalink
fix lint error
Browse files Browse the repository at this point in the history
  • Loading branch information
Laurawly committed Mar 15, 2019
1 parent d0045e0 commit b66ff61
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion topi/python/topi/cuda/nms.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,8 @@ def get_valid_counts_ir(data, flag, idx, valid_count, out):
base_idx = i * num_anchors * elem_length
with ib.if_scope(flag[tid] > 0):
with ib.for_range(0, elem_length) as k:
out[base_idx + (idx[tid] - 1) * elem_length + k] = data[base_idx + j * elem_length + k]
out[base_idx + (idx[tid] - 1) * elem_length + k] =\
data[base_idx + j * elem_length + k]
valid_count[i] = idx[i * num_anchors + num_anchors - 1]

return ib.get()
Expand Down

0 comments on commit b66ff61

Please sign in to comment.