pytorch/torch/amp
CaoE c47d2b8035 Add Half support for CPU autocast on eager mode (#112484)
Add Half support for CPU autocast on eager mode since common operators have Half support on CPU.
https://github.com/pytorch/pytorch/issues/96093.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/112484
Approved by: https://github.com/leslie-fang-intel, https://github.com/ezyang
2023-11-21 20:08:28 +00:00
..
__init__.py Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
autocast_mode.py Add Half support for CPU autocast on eager mode (#112484) 2023-11-21 20:08:28 +00:00