Yu, Guangye 19a83eacb5 add new API torch.amp.is_autocast_available (#124938) ...

# Motivation
expose `torch._is_autocast_available` to `torch.amp.is_autocast_available` as a public api.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/124938
Approved by: https://github.com/albanD

2024-04-26 08:45:20 +00:00

..

__init__.py

add new API torch.amp.is_autocast_available (#124938)

2024-04-26 08:45:20 +00:00

autocast_mode.py

add new API torch.amp.is_autocast_available (#124938)

2024-04-26 08:45:20 +00:00

grad_scaler.py

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

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