Revert embedding thrust->cub migration

[ngimel]

Fixes #63427

[facebook-github-bot]

🔗 Helpful links

• 🧪 See artifacts and rendered test results at hud.pytorch.org/pr/63451
• 📄 Preview docs built from this PR

💊 CI failures summary and remediations

As of commit a5a704f (more details on the Dr. CI page):

---

• 1/1 failures possibly* introduced in this PR
  • 1/1 non-scanned failure(s)

---

ci.pytorch.org: 1 failed

• Failed: pr/pytorch-linux-bionic-rocm4.2-py3.6

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

[codecov]

Codecov Report

Merging #63451 (a5a704f) into master (dcf90b7) will increase coverage by 14.59%.
The diff coverage is n/a.

@@             Coverage Diff             @@
##           master   #63451       +/-   ##
===========================================
+ Coverage   59.86%   74.46%   +14.59%     
===========================================
  Files        2068     2117       +49     
  Lines      210331   212216     +1885     
===========================================
+ Hits       125921   158028    +32107     
+ Misses      84410    54188    -30222     

[facebook-github-bot]

@ngimel has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

[facebook-github-bot]

@ngimel merged this pull request in ce6fe50.