pytorch
61f946bb - don't copy indices to the self device in dispatch_index (#59059)

Commit
3 years ago
don't copy indices to the self device in dispatch_index (#59059) Summary: Let index/index_put implementation in aten take care of moving the indices to the correct device, don't make python wrapper do that. Pull Request resolved: https://github.com/pytorch/pytorch/pull/59059 Reviewed By: mruberry Differential Revision: D28750562 Pulled By: ngimel fbshipit-source-id: 2f2b5f875733898f1c0b30b544c89808f91e4a6f
Author
Natalia Gimelshein
Parents
Loading