pytorch
d90911ad - fix AdaptiveAveragePooling crash problem for non support input (#51443)

Commit
3 years ago
fix AdaptiveAveragePooling crash problem for non support input (#51443) Summary: For none support input, we should not do check in a parallel region, this PR will first do the dtype check, and then do parallel for. Fixes https://github.com/pytorch/pytorch/issues/51352. Pull Request resolved: https://github.com/pytorch/pytorch/pull/51443 Reviewed By: izdeby Differential Revision: D26305584 Pulled By: ngimel fbshipit-source-id: 6faa3148af5bdcd7246771c0ecb4db2b31ac82c6
Author
Parents
Loading