Does sort have GPU enhancements?

1

Hi,

Does the tensor.sort function have a GPU implementation or just a CPU one?

2

Definitely gpu accelerated: https://github.com/pytorch/pytorch/blob/dc0faab18dd5296b66dfb00776253062def87fdf/aten/src/THC/generic/THCTensorSort.cu#L279