projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 94d6215) | patch
[Reland] Embedding thrust->cub migration (#63806)
authorXiang Gao <qasdfgtyuiop@gmail.com>
Tue, 24 Aug 2021 16:24:50 +0000 (09:24 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 24 Aug 2021 16:30:32 +0000 (09:30 -0700)
commit227cb268bccd22feb8aa8651773a202ec1e09c7f
tree890852740d321cf02b87af87a98b880317350fcatree | snapshot
parent94d621584a8d2780252546aa787aab23203221b2commit | diff
[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
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
torch/testing/_internal/common_nn.py diff | blob | history
Domain: Machine Learning / ML Framework;