pytorch
8805093e - use long index type for index_add_cuda deterministic path (#59254)

Commit
3 years ago
use long index type for index_add_cuda deterministic path (#59254) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59254 index_add can take int or long index tensor whereas index_put only takes long indices tensor. In the deterministic path of index_add_cuda, we use index_put. Hence we better convert index tensor to long. Test Plan: buck test mode/opt //caffe2/test:torch_cuda -- test_index_add_deterministic ✓ ListingSuccess: caffe2/test:torch_cuda - main (14.748) ✓ Pass: caffe2/test:torch_cuda - test_index_add_deterministic_cuda (test_torch.TestTorchDeviceTypeCUDA) (27.717) ✓ Pass: caffe2/test:torch_cuda - main (27.717) Reviewed By: ngimel Differential Revision: D28804038 fbshipit-source-id: de12932a7738f2805f3bceb3ec024497625bce6a
Author
Parents
Loading