Roy Hvaara 4b83302585 [MPS] Update error message for supported autocast type (#139192) ...

Autocast in MPS currently only supports dtype of `torch.float16`. This PR updates the error message to reflect this.

This PR was created using [Copilot Workspace](https://copilot-workspace.githubnext.com/pytorch/pytorch/issues/139190?shareId=5b510fda-380c-4e86-8e91-6b67a078f180) with no human input other than clicking buttons.

Fixes #139190

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139192
Approved by: https://github.com/malfet

2024-10-30 16:48:29 +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

[MPS] Update error message for supported autocast type (#139192)

2024-10-30 16:48:29 +00:00

grad_scaler.py

Add support to GradScaler for respecting an already set grad_scale value (#123429)

2024-06-27 22:40:54 +00:00