Kulin Seth 144fde4fd2 [MPS] Add support for autocast in MPS (#99272) ... Fixes https://github.com/pytorch/pytorch/issues/88415 Need to run inductor/test_cpu_select_algorithm Pull Request resolved: https://github.com/pytorch/pytorch/pull/99272 Approved by: https://github.com/malfet Co-authored-by: Siddharth Kotapati <skotapati@apple.com> Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com> Co-authored-by: Roy Hvaara <roy@lightyear.no>		2024-09-05 23:23:17 +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] Add support for autocast in MPS (#99272)	2024-09-05 23:23:17 +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