pytorch

80826776 - Prevent grad scale from overflowing (#98876)

Prevent grad scale from overflowing (#98876) Fixes #98828 by capping the growth in the kernel Pull Request resolved: https://github.com/pytorch/pytorch/pull/98876 Approved by: https://github.com/ngimel