Revert embedding thrust->cub migration (#63451)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/63427
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63451
Reviewed By: mruberry
Differential Revision: D30398482
Pulled By: ngimel
fbshipit-source-id: e153786d204215555a6571688eabae712facad7e
