Fix a cuda max_pool3d issue, do multiplication in int64 (#52828)
Summary:
Fix https://github.com/pytorch/pytorch/issues/52822
- [x] benchmark
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52828
Reviewed By: mrshenli
Differential Revision: D26866674
Pulled By: heitorschueroff
fbshipit-source-id: bd8276dd70316a767dc6e1991c1259f1f0b390b2