pytorch
8596ac18 - deterministic code path for gather_backward for dim = 1 (#55573)

Commit
3 years ago
deterministic code path for gather_backward for dim = 1 (#55573) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55573 provide a deterministic code path for gather_backward when dim = 1 Test Plan: buck test //caffe2/test:torch -- test_gather_backward ✓ Pass: caffe2/test:torch - test_gather_backward_one_dim (test_torch.TestTorch) (1.099) ✓ Pass: caffe2/test:torch - test_gather_backward_deterministic_path (test_torch.TestTorch) (1.166) test on GPU buck test mode/opt //caffe2/test:torch_cuda -- test_gather_backward_deterministic Started reporting to test run: https://www.internalfb.com/intern/testinfra/testrun/1407375070421778 ✓ ListingSuccess: caffe2/test:torch_cuda - main (7.484) ✓ Pass: caffe2/test:torch_cuda - test_gather_backward_deterministic_path_cuda (test_torch.TestTorchDeviceTypeCUDA) (26.145) ✓ Pass: caffe2/test:torch_cuda - main (26.145) Summary Pass: 2 ListingSuccess: 1 Reviewed By: ngimel Differential Revision: D27632008 fbshipit-source-id: ec27475332a3b36360cc014193256c21cba77d63
Author
Parents
Loading