projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 9920358) | patch
Revert embedding thrust->cub migration (#63451)
authorNatalia Gimelshein <ngimel@fb.com>
Thu, 19 Aug 2021 20:00:08 +0000 (13:00 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 19 Aug 2021 20:03:33 +0000 (13:03 -0700)
commitce6fe50158c631f7f54b6df3ac91632cd41d48ea
treeb396cf428142c3ebf0296bc2eaf871f6f6a9c98dtree | snapshot
parent99203580a9e2bb468a9d814f36b6c6a4c13fbed4commit | diff
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
aten/src/ATen/cuda/cub.cuh diff | blob | history
aten/src/ATen/native/cuda/Embedding.cu diff | blob | history
aten/src/ATen/native/cuda/EmbeddingBackwardKernel.cuh diff | blob | history
aten/src/ATen/native/cuda/Indexing.cu diff | blob | history
aten/src/ATen/native/cuda/LegacyThrustHelpers.cu diff | blob | history
aten/src/ATen/native/cuda/Randperm.cu diff | blob | history
aten/src/ATen/native/cuda/UniqueCub.cu diff | blob | history
Domain: Machine Learning / ML Framework;