pytorch

0b93afb1 - add amp tests (#85434)

add amp tests (#85434) Pull Request resolved: https://github.com/pytorch/pytorch/pull/85434 Approved by: https://github.com/ngimel