[Reland] Embedding thrust->cub migration (#63806)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/63427
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63806
Reviewed By: bdhirsh
Differential Revision: D30498255
Pulled By: ngimel
fbshipit-source-id: 78b7085a92a168cf0163f53dcb712bac922f5235
