pytorch
9e79a8a5 - [iOS GPU][MaskRCNN] Force the temporaryImage to become static when doing synchronization (#60155)

Commit
3 years ago
[iOS GPU][MaskRCNN] Force the temporaryImage to become static when doing synchronization (#60155) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60155 For intermediate tensors, we need to convert them to static images when doing GPU -> CPU synchronization. ghstack-source-id: 131540760 Test Plan: - CI - buck test pp-macos Reviewed By: SS-JIA Differential Revision: D29126278 fbshipit-source-id: cd50b5f104e0161ec7fcfcc2c51785f241e48704
Author
Parents
Loading