pytorch
89d0e31f - [torch][repeat_interleave] Remove stream sync when output_size is given for scalar repeats (#60965)

Commit
3 years ago
[torch][repeat_interleave] Remove stream sync when output_size is given for scalar repeats (#60965) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60965 Same as title. Simple change on tensor creation. Test Plan: Rely on existing signals and verify manually that sync is not happening. Reviewed By: ngimel Differential Revision: D29461773 fbshipit-source-id: 21d6ebfba08449da39fc7f109958f6c6978a4f32
Author
Serhat Yilmaz
Parents
Loading