pytorch
b5b7d6a3 - EmbeddingBackward exclusive_scan thrust->cub (#66566)

Commit
3 years ago
EmbeddingBackward exclusive_scan thrust->cub (#66566) Summary: Fixes #{issue number} Pull Request resolved: https://github.com/pytorch/pytorch/pull/66566 Reviewed By: H-Huang Differential Revision: D31637660 Pulled By: ngimel fbshipit-source-id: 8093432bb9a9b902bb6bab7da221f0bcd7e9fb34
Author
Parents
Loading