diff --git a/python/tvm/topi/cuda/sort.py b/python/tvm/topi/cuda/sort.py index 47684b8f778e..b8873e0b546b 100644 --- a/python/tvm/topi/cuda/sort.py +++ b/python/tvm/topi/cuda/sort.py @@ -317,8 +317,8 @@ def argsort_nms_thrust(data, valid_count, axis=-1, is_ascend=1, dtype="float32") def sort(data, axis=-1, is_ascend=1): - """Performs sorting along the given axis and returns an array of indicies - having same shape as an input array that index data in sorted order. + """Performs sorting along the given axis and returns an array of + sorted values with teh same shape as the input data. Parameters ----------