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