pytorch

eff74ed7 - [AMP] Use generic autocast in example, specify dtype (#79579)

[AMP] Use generic autocast in example, specify dtype (#79579) CC @mruberry @ptrblck Pull Request resolved: https://github.com/pytorch/pytorch/pull/79579 Approved by: https://github.com/mruberry, https://github.com/ngimel