pytorch
3f656ad7 - [CUDA] Do accumulation for Adaptive Average Pooling in `opmath_t` (#99378)

Commit
1 year ago
[CUDA] Do accumulation for Adaptive Average Pooling in `opmath_t` (#99378) Fix for an issue surfaced from the discuss forum: https://discuss.pytorch.org/t/adaptiveavgpool2d-causes-some-data-to-contain-inf/177420 CC @ptrblck @ngimel Pull Request resolved: https://github.com/pytorch/pytorch/pull/99378 Approved by: https://github.com/ngimel
Author
eqy eqy
Committer
Parents
Loading