Bring back old algorithm for sorting on small number of segments (#64127)
authorXiang Gao <qasdfgtyuiop@gmail.com>
Mon, 30 Aug 2021 19:25:29 +0000 (12:25 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 30 Aug 2021 19:30:50 +0000 (12:30 -0700)
commita8ffe81b2c3123926354b4ec2001693b38daa80d
treeb18d5c67d2f0e571d465794ebcec7d755dc0b067
parentd37636901ed1c65c1f8b68e36e37e59eb503c554
Bring back old algorithm for sorting on small number of segments (#64127)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/63456
The code was copy-pasted from the previous commit without modification.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/64127

Reviewed By: mruberry

Differential Revision: D30632090

Pulled By: ngimel

fbshipit-source-id: 58bbdd9b0423f01d4e65e2ec925ad9a3f88efc9b
aten/src/ATen/native/cuda/Sort.cu