pytorch
4ca4640b - [torch][repeat_interleave] remove stream syncronization if output size is given (#58417)

Commit
3 years ago
[torch][repeat_interleave] remove stream syncronization if output size is given (#58417) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58417 Same as title. Test Plan: Rely on CI signal. Update unit test to exercise new code path as well. Reviewed By: ngimel Differential Revision: D28482927 fbshipit-source-id: 3ec8682810ed5c8547b1e8d3869924480ce63dcd
Author
Serhat Yilmaz
Parents
Loading