Tom McTiernan aa11628576 Issue warning with reference to user code rather than torch (#155112) ... Re-raising of #129959 as that was closed. Warning message before: ``` /home/admin/.local/share/hatch/env/virtual/toms-project-1/Qv9k_r_5/dev/lib/python3.10/site-packages/torch/cuda/amp/grad_scaler.py:120: UserWarning: torch.cuda.amp.GradScaler is enabled, but CUDA is not available. Disabling. ``` Warning message after: ``` /path/to/my/code:91: UserWarning: torch.cuda.amp.GradScaler is enabled, but CUDA is not available. Disabling. ``` Helps the user find where the issue stems from in their code. What do you think? (Looks like "skip_file_prefixes" is not available until Python 3.12 minimum...) Pull Request resolved: https://github.com/pytorch/pytorch/pull/155112 Approved by: https://github.com/Skylion007, https://github.com/cyyever		2025-07-14 05:24:23 +00:00
..
__init__.py
autocast_mode.py	[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)	2025-07-03 18:56:07 +00:00
grad_scaler.py	Issue warning with reference to user code rather than torch (#155112)	2025-07-14 05:24:23 +00:00