saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/torch/amp
History
Simon Mahns 6939a56e13 [autocast][pytorch] Support autocast for MTIA (#145627) 
Summary: Add autocast support to MTIA

Reviewed By: egienvalue

Differential Revision: D68572548

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145627
Approved by: https://github.com/egienvalue
2025-01-25 03:24:59 +00:00
..
__init__.py
autocast_mode.py [autocast][pytorch] Support autocast for MTIA (#145627) 2025-01-25 03:24:59 +00:00
grad_scaler.py