Tugsbayasgalan (Tugsuu) Manlaibaatar 2ac45c2752 Fix autocast context manager when there is exception (#159565) ...

Summary: When exception occurs inside context manager, we need to either return False OR properly propagage exceptions via __exit__(exc_type, exc_val). But previously while tracing, we don't actually run the exit node so we end up swallowing the exception in a very weird way as outlined in https://github.com/pytorch/pytorch/issues/153202. This PR fixes it

Test Plan:
new test case

Rollback Plan:

Differential Revision: D79348382

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159565
Approved by: https://github.com/zou3519, https://github.com/yushangdi

2025-08-01 02:12:24 +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

Fix autocast context manager when there is exception (#159565)

2025-08-01 02:12:24 +00:00

grad_scaler.py

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

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