PyTorch MergeBot 9420944033 Revert "[AMP][Refactor] Simplify dtype support logic in autocast context manager (#163446)" ...

This reverts commit 960b0d5f0d0efb1f1962bddcf62e2a698e26edd2.

Reverted https://github.com/pytorch/pytorch/pull/163446 on behalf of https://github.com/izaitsevfb due to breaks autocast tests on linux and mac ([comment](https://github.com/pytorch/pytorch/pull/163446#issuecomment-3390688642))

2025-10-10 15:12:46 +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

Revert "[AMP][Refactor] Simplify dtype support logic in autocast context manager (#163446)"

2025-10-10 15:12:46 +00:00

grad_scaler.py

Issue warning with reference to user code rather than torch (#155112)

2025-07-14 05:24:23 +00:00