Port `sort` kernel to structured kernels. (#62391)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62391
Tracking issue: #55070
Test Plan: Imported from OSS
Reviewed By: bhosmer
Differential Revision: D30903992
Pulled By: bdhirsh
fbshipit-source-id: 52687aa2483c101056825433d39d69c60b829c62
