Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
mbrookhart committed Dec 15, 2020
1 parent f2af723 commit 8c6b03b
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions python/tvm/topi/cuda/sort.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,10 +26,6 @@
from .. import tag


def ceil_div(a, b):
return (a + b - 1) // b


def swap(arr, axis):
""" swap arr[axis] and arr[-1] """
return arr[:axis] + [arr[-1]] + arr[axis + 1 : -1] + [arr[axis]]
Expand Down Expand Up @@ -145,7 +141,6 @@ def ceil_div(a, b):
ib.scope_attr(bz, "thread_extent", nthread_bz)
idx = (by * shape[axis] + tid) * axis_mul_after + bz
with ib.if_scope(tid < shape[axis]):
idx = (by * shape[axis] + tid) * axis_mul_after + bz
values_out[idx] = data[idx]
if indices_out is not None:
indices_out[idx] = tvm.tir.generic.cast(tid, indices_out.dtype)
Expand Down

0 comments on commit 8c6b03b

Please sign in to comment.