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

generalize custom_fwd&custom_bwd to be device-agnostic (#126531)

2024-05-25 06:48:16 +00:00

autocast_mode.py

[autocast][pytorch] Support autocast for MTIA (#145627)

2025-01-25 03:24:59 +00:00

grad_scaler.py

PEP585 update - mostly toplevels (#145178)

2025-01-22 02:21:14 +00:00