pytorch
846c8d94 - mark embedding backward non-deterministic for max mode rather than all reducing modes (#55574)

Commit
3 years ago
mark embedding backward non-deterministic for max mode rather than all reducing modes (#55574) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55574 nn.EmbeddingBag backward is non-deterministic when reducing_mode = Max and on GPU, reducing mode Mean and Sum should be deterministic Test Plan: NA Reviewed By: ngimel Differential Revision: D27633832 fbshipit-source-id: 50786ed8522f1aae27442f5f244a65eab8000b06
Author
Parents
Loading