pytorch
8a450067 - enable deterministic path for index_copy_cuda with index_put (#58144)

Commit
4 years ago
enable deterministic path for index_copy_cuda with index_put (#58144) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58144 reland D28291041 (https://github.com/pytorch/pytorch/commit/14badd9929e3fb6aad74983c6ba82658be6ab109), which was reverted due to a type error from Tuple[torch.Tensor], seems that mypy requires Tuple[torch.Tensor, torch.Tensor, torch.Tensor] Test Plan: buck test mode/opt //caffe2/test:torch_cuda -- test_index_copy_deterministic ✓ ListingSuccess: caffe2/test:torch_cuda - main (9.229) ✓ Pass: caffe2/test:torch_cuda - test_index_copy_deterministic_cuda (test_torch.TestTorchDeviceTypeCUDA) (25.750) ✓ Pass: caffe2/test:torch_cuda - main (25.750) Reviewed By: ngimel Differential Revision: D28383178 fbshipit-source-id: 38896fd6ddd670cfcce36e079aee7ad52adc2a28
Author
Parents
Loading