remove syncs in one_hot (#57902)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/55579
Now on cuda one-hot relies on device-side asserts thrown by scatter
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57902
Reviewed By: bdhirsh
Differential Revision: D28328698
Pulled By: ngimel
fbshipit-source-id: 1cd13e2c123c733cde7dbe4cbe6ff5335063bb70
Author
Natalia Gimelshein
