Aaron Gokaslan 9c4fc5fa34 [BE][Ez]: Fix minor potential perf regression from #123960 (#124013) ...

The `non_blocking` arg here is useless if the values are all eagerly consumed, so revert the change.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/124013
Approved by: https://github.com/ezyang

2024-04-15 16:51:45 +00:00

..

__init__.py

add GradScaler on CPU (#109993)

2024-01-29 23:42:35 +00:00

autocast_mode.py

Add Half support for CPU autocast on eager mode (#112484)

2023-11-21 20:08:28 +00:00

grad_scaler.py

[BE][Ez]: Fix minor potential perf regression from #123960 (#124013)

2024-04-15 16:51:45 +00:00